train.yaml 1.1 KB

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