train.yaml 1.1 KB

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