16 lines
340 B
Python
16 lines
340 B
Python
|
# checkpoint saving
|
||
|
|
checkpoint_config = dict(interval=1)
|
||
|
|
# yapf:disable
|
||
|
|
log_config = dict(
|
||
|
|
interval=100,
|
||
|
|
hooks=[
|
||
|
|
dict(type='TextLoggerHook'),
|
||
|
|
# dict(type='TensorboardLoggerHook')
|
||
|
|
])
|
||
|
|
# yapf:enable
|
||
|
|
dist_params = dict(backend='nccl')
|
||
|
|
log_level = 'INFO'
|
||
|
|
load_from = None
|
||
|
|
resume_from = None
|
||
|
|
workflow = [('train', 1)]
|