train.yaml 1.3 KB

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