mmpretrain/configs/_base_/default_runtime.py

38 lines
856 B
Python
Raw Normal View History

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
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
log_level = 'INFO'
2022-05-23 16:43:23 +08:00
# Load from weight
load_from = None
2022-05-23 16:43:23 +08:00
# resume training
resume = False