train.yaml 336 B

123456789101112131415161718
  1. io:
  2. logdir: logs/
  3. datadir: D:\python\PycharmProjects\data
  4. # datadir: I:\datasets\wirenet_1000
  5. resume_from:
  6. num_workers: 8
  7. tensorboard_port: 6000
  8. validation_interval: 300
  9. batch_size: 4
  10. batch_size_eval: 2
  11. optim:
  12. name: Adam
  13. lr: 4.0e-4
  14. amsgrad: True
  15. weight_decay: 1.0e-4
  16. max_epoch: 90000000
  17. lr_decay_epoch: 10