EasyCV/configs/base.py
Chen Jiayu 9517bb80ff
add dn_detr algo (#119)
support dn-detr-r50: 44.39
refactor set_criterion
2022-07-27 15:06:06 +08:00

19 lines
419 B
Python

train_cfg = {}
test_cfg = {}
optimizer_config = dict() # grad_clip, coalesce, bucket_size_mb
# yapf:disable
log_config = dict(
interval=50,
hooks=[
dict(type='TextLoggerHook'),
# dict(type='TensorboardLoggerHook')
])
# yapf:enable
# runtime settings
dist_params = dict(backend='nccl')
cudnn_benchmark = False
log_level = 'INFO'
load_from = None
resume_from = None
workflow = [('train', 1)]