Modify config relates to logger and checkpoint

configs/_base_/default_runtime.py

@@ -8,7 +8,7 @@ default_hooks = dict(
     timer=dict(type='IterTimerHook'),
 
     # logger configure
-    logger=dict(type='LoggerHook', interval=50),
+    logger=dict(type='LoggerHook', interval=100),
 
     # Parameter Scheduler
     param_scheduler=dict(type='ParamSchedulerHook'),

configs/lenet/lenet5_mnist.py

@@ -40,15 +40,11 @@ optimizer_config = dict(grad_clip=None)
 # learning policy
 param_scheduler = dict(
     type='MultiStepLR', by_epoch=True, milestones=[15], gamma=0.1)
-# checkpoint saving
-checkpoint_config = dict(interval=1)
 # yapf:disable
-log_config = dict(
-    interval=150,
-    hooks=[
-        dict(type='TextLoggerHook'),
-        # dict(type='TensorboardLoggerHook')
-    ])
+default_hooks = dict(
+    logger=dict(type='LoggerHook', interval=150),
+    checkpoint=dict(type='CheckpointHook', interval=1)
+)
 # yapf:enable
 
 # train, val, test setting

configs/resnet/resnet50_8xb8_cub.py

@@ -13,7 +13,8 @@ model = dict(
             type='Pretrained', checkpoint=checkpoint, prefix='backbone')),
     head=dict(num_classes=200, ))
 
-log_config = dict(interval=20)  # log every 20 intervals
-
-checkpoint_config = dict(
-    interval=1, max_keep_ckpts=3)  # save last three checkpoints
+default_hooks = dict(
+    # log every 20 intervals
+    logger=dict(type='LoggerHook', interval=20),
+    # save last three checkpoints
+    checkpoint=dict(type='CheckpointHook', interval=1, max_keep_ckpts=3))

configs/swin_transformer/swin-large_8xb8_cub_384px.py

@@ -31,7 +31,8 @@ optimizer = dict(
     paramwise_cfg=paramwise_cfg)
 optimizer_config = dict(grad_clip=dict(max_norm=5.0), _delete_=True)
 
-log_config = dict(interval=20)  # log every 20 intervals
-
-checkpoint_config = dict(
-    interval=1, max_keep_ckpts=3)  # save last three checkpoints
+default_hooks = dict(
+    # log every 20 intervals
+    logger=dict(type='LoggerHook', interval=20),
+    # save last three checkpoints
+    checkpoint=dict(type='CheckpointHook', interval=1, max_keep_ckpts=3))

configs/vision_transformer/vit-base-p16_ft-4xb544-ipu_in1k.py

@@ -112,6 +112,6 @@ runner = dict(
     options_cfg=options_cfg,
     max_iters=5000)
 
-checkpoint_config = dict(interval=1000)
+default_hooks = dict(checkpoint=dict(type='CheckpointHook', interval=1000))
 
 fp16 = dict(loss_scale=256.0, velocity_accum_type='half', accum_type='half')