[Fix] Default hooks in cgnet config (#1897)

This commit is contained in:
Miao Zheng 2022-08-09 20:26:05 +08:00 committed by GitHub
parent 7e83c1dd97
commit 78b08650b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 2 deletions

View File

@ -23,7 +23,11 @@ train_cfg = dict(
val_cfg = dict(type='ValLoop')
test_cfg = dict(type='TestLoop')
default_hooks = dict(
checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=4000))
timer=dict(type='IterTimerHook'),
logger=dict(type='LoggerHook', interval=50),
param_scheduler=dict(type='ParamSchedulerHook'),
checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=4000),
sampler_seed=dict(type='DistSamplerSeedHook'))
crop_size = (512, 1024)
data_preprocessor = dict(size=crop_size)

View File

@ -23,7 +23,11 @@ train_cfg = dict(
val_cfg = dict(type='ValLoop')
test_cfg = dict(type='TestLoop')
default_hooks = dict(
checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=4000))
timer=dict(type='IterTimerHook'),
logger=dict(type='LoggerHook', interval=50),
param_scheduler=dict(type='ParamSchedulerHook'),
checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=4000),
sampler_seed=dict(type='DistSamplerSeedHook'))
crop_size = (680, 680)
data_preprocessor = dict(size=crop_size)