You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
checkpoint_config = dict(interval=1)
|
|
|
|
# yapf:disable
|
|
|
|
log_config = dict(
|
|
|
|
interval=50,
|
|
|
|
hooks=[
|
|
|
|
dict(type='TextLoggerHook'),
|
|
|
|
# dict(type='TensorboardLoggerHook')
|
|
|
|
])
|
|
|
|
# yapf:enable
|
|
|
|
custom_hooks = [dict(type='NumClassCheckHook')]
|
|
|
|
|
|
|
|
dist_params = dict(backend='nccl')
|
|
|
|
log_level = 'INFO'
|
|
|
|
load_from = None
|
|
|
|
resume_from = None
|
|
|
|
workflow = [('train', 1)]
|
|
|
|
|
|
|
|
# disable opencv multithreading to avoid system being overloaded
|
|
|
|
opencv_num_threads = 0
|
|
|
|
# set multi-process start method as `fork` to speed up the training
|
|
|
|
mp_start_method = 'fork'
|
|
|
|
|
|
|
|
# Default setting for scaling LR automatically
|
|
|
|
# - `enable` means enable scaling LR automatically
|
|
|
|
# or not by default.
|
|
|
|
# - `base_batch_size` = (8 GPUs) x (2 samples per GPU).
|
|
|
|
auto_scale_lr = dict(enable=False, base_batch_size=16)
|