mirror of
https://huggingface.co/spaces/H-Liu1997/TANGO
synced 2026-04-22 16:17:09 +08:00
H-Liu1997
21 lines
537 B
Python
21 lines
537 B
Python
checkpoint_config = dict(interval=10)
|
|
|
|
log_config = dict(
|
|
interval=50,
|
|
hooks=[
|
|
dict(type='TextLoggerHook'),
|
|
# dict(type='TensorboardLoggerHook')
|
|
# dict(type='PaviLoggerHook') # for internal services
|
|
])
|
|
|
|
log_level = 'INFO'
|
|
load_from = None
|
|
resume_from = None
|
|
dist_params = dict(backend='nccl')
|
|
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'
|