train.yaml 1011 B

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