train.yaml 894 B

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