train.yaml 293 B

12345678910111213141516
  1. io:
  2. logdir: logs/
  3. datadir: J:/datasets/wirenet_rgb_gray
  4. # datadir: I:\datasets\wirenet_1000
  5. resume_from:
  6. num_workers: 8
  7. tensorboard_port: 6000
  8. validation_interval: 300
  9. optim:
  10. name: Adam
  11. lr: 4.0e-4
  12. amsgrad: True
  13. weight_decay: 1.0e-4
  14. max_epoch: 1000
  15. lr_decay_epoch: 10