default_runtime.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. default_scope = 'mmyolo'
  2. custom_imports = dict(imports=['models', 'datasets'])
  3. # hooks
  4. default_hooks = dict(
  5. timer=dict(type='IterTimerHook'),
  6. logger=dict(type='LoggerHook', interval=50),
  7. param_scheduler=dict(type='ParamSchedulerHook'),
  8. checkpoint=dict(type='CheckpointHook', interval=10, max_keep_ckpts=3),
  9. sampler_seed=dict(type='DistSamplerSeedHook'),
  10. visualization=dict(type='mmpose.PoseVisualizationHook', enable=False),
  11. )
  12. # multi-processing backend
  13. env_cfg = dict(
  14. cudnn_benchmark=False,
  15. mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
  16. dist_cfg=dict(backend='nccl'),
  17. )
  18. # visualizer
  19. vis_backends = [dict(type='LocalVisBackend')]
  20. visualizer = dict(
  21. type='mmpose.PoseLocalVisualizer',
  22. vis_backends=vis_backends,
  23. name='visualizer')
  24. # logger
  25. log_processor = dict(
  26. type='LogProcessor', window_size=50, by_epoch=True, num_digits=6)
  27. log_level = 'INFO'
  28. load_from = None
  29. resume = False
  30. # file I/O backend
  31. file_client_args = dict(backend='disk')
  32. # training/validation/testing progress
  33. train_cfg = dict()
  34. val_cfg = dict(type='ValLoop')
  35. test_cfg = dict(type='TestLoop')