default_runtime.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. default_scope = 'mmpose'
  2. # hooks
  3. default_hooks = dict(
  4. timer=dict(type='IterTimerHook'),
  5. logger=dict(type='LoggerHook', interval=50),
  6. param_scheduler=dict(type='ParamSchedulerHook'),
  7. checkpoint=dict(type='CheckpointHook', interval=10),
  8. sampler_seed=dict(type='DistSamplerSeedHook'),
  9. visualization=dict(type='PoseVisualizationHook', enable=False),
  10. )
  11. # custom hooks
  12. custom_hooks = [
  13. # Synchronize model buffers such as running_mean and running_var in BN
  14. # at the end of each epoch
  15. dict(type='SyncBuffersHook')
  16. ]
  17. # multi-processing backend
  18. env_cfg = dict(
  19. cudnn_benchmark=False,
  20. mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
  21. dist_cfg=dict(backend='nccl'),
  22. )
  23. # visualizer
  24. vis_backends = [
  25. dict(type='LocalVisBackend'),
  26. # dict(type='TensorboardVisBackend'),
  27. # dict(type='WandbVisBackend'),
  28. ]
  29. visualizer = dict(
  30. type='PoseLocalVisualizer', vis_backends=vis_backends, name='visualizer')
  31. # logger
  32. log_processor = dict(
  33. type='LogProcessor', window_size=50, by_epoch=True, num_digits=6)
  34. log_level = 'INFO'
  35. load_from = None
  36. resume = False
  37. # file I/O backend
  38. backend_args = dict(backend='local')
  39. # training/validation/testing progress
  40. train_cfg = dict(by_epoch=True)
  41. val_cfg = dict()
  42. test_cfg = dict()