2022-05-23 16:43:23 +08:00
|
|
|
default_scope = 'mmcls'
|
|
|
|
|
|
|
|
default_hooks = dict(
|
|
|
|
# optimizer configure
|
|
|
|
optimizer=dict(type='OptimizerHook', grad_clip=None),
|
|
|
|
|
|
|
|
# record the time to load data and the time it takes to iterate once
|
|
|
|
timer=dict(type='IterTimerHook'),
|
|
|
|
|
|
|
|
# logger configure
|
2022-05-24 17:23:05 +08:00
|
|
|
logger=dict(type='LoggerHook', interval=100),
|
2022-05-23 16:43:23 +08:00
|
|
|
|
|
|
|
# Parameter Scheduler
|
|
|
|
param_scheduler=dict(type='ParamSchedulerHook'),
|
|
|
|
|
|
|
|
# checkpoint saving
|
|
|
|
checkpoint=dict(type='CheckpointHook', interval=1),
|
|
|
|
|
|
|
|
# Sampler for distributed training
|
|
|
|
sampler_seed=dict(type='DistSamplerSeedHook'),
|
|
|
|
)
|
|
|
|
|
|
|
|
# Environment configure
|
|
|
|
env_cfg = dict(
|
|
|
|
cudnn_benchmark=False,
|
|
|
|
mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
|
|
|
|
dist_cfg=dict(backend='nccl'),
|
|
|
|
)
|
|
|
|
|
|
|
|
# Log level configuration
|
2020-07-09 23:59:10 +08:00
|
|
|
log_level = 'INFO'
|
2022-05-23 16:43:23 +08:00
|
|
|
|
|
|
|
# Load from weight
|
2020-07-09 23:59:10 +08:00
|
|
|
load_from = None
|
2022-05-23 16:43:23 +08:00
|
|
|
|
|
|
|
# resume training
|
|
|
|
resume = False
|