train.yaml 258 B

123456789101112131415
  1. io:
  2. logdir: logs/
  3. datadir: D:\python\PycharmProjects\data
  4. resume_from:
  5. num_workers: 8
  6. tensorboard_port: 6000
  7. validation_interval: 300
  8. optim:
  9. name: Adam
  10. lr: 4.0e-4
  11. amsgrad: True
  12. weight_decay: 1.0e-4
  13. max_epoch: 1000
  14. lr_decay_epoch: 10