train.yaml 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. io:
  2. logdir: train_results
  3. <<<<<<< HEAD
  4. datadir: /data/share/rlq/datasets/250718caisegangban
  5. # datadir: /data/share/zjh/Dataset_correct_xanylabel_tiff
  6. =======
  7. # datadir: /data/share/rlq/datasets/250718caisegangban
  8. >>>>>>> 55050f283a5d85b7c3731e296d70483d8f7258e2
  9. # datadir: /data/share/rlq/datasets/singepoint_Dataset0709_2
  10. datadir: \\192.168.50.222/share/zyh/arc/a_dataset
  11. # datadir: \\192.168.50.222/share/rlq/datasets/singepoint_Dataset0709_2
  12. # datadir: \\192.168.50.222/share/rlq/datasets/250718caisegangban
  13. data_type: rgb
  14. # datadir: D:\python\PycharmProjects\data_20250223\0423_
  15. # datadir: I:\datasets\wirenet_1000
  16. tensorboard_port: 6000
  17. validation_interval: 300
  18. train_params:
  19. resume_from:
  20. num_workers: 8
  21. batch_size: 3
  22. max_epoch: 8000000
  23. # augmentation: True
  24. augmentation: False
  25. optim:
  26. name: Adam
  27. lr: 4.0e-4
  28. amsgrad: True
  29. weight_decay: 1.0e-4
  30. lr_decay_epoch: 10
  31. # 冻结参数
  32. freeze_params:
  33. backbone: False,
  34. rpn: False,
  35. roi_heads:
  36. box_head: False,
  37. box_predictor: False,
  38. line_head: False,
  39. line_predictor:
  40. fc1: False,
  41. fc2:
  42. 0: False,
  43. 2: False,
  44. 4: False