| 12345678910111213141516171819202122232425262728293031323334353637383940 |
- io:
- logdir: train_results
- datadir: /data/share/zyh/202507/a_dataset
- data_type: rgb
- # datadir: D:\python\PycharmProjects\data_20250223\0423_
- # datadir: I:\datasets\wirenet_1000
- tensorboard_port: 6000
- validation_interval: 300
- train_params:
- resume_from:
- num_workers: 8
- batch_size: 4
- max_epoch: 80000
- augmentation: True
- optim:
- name: Adam
- lr: 4.0e-4
- amsgrad: True
- weight_decay: 1.0e-4
- lr_decay_epoch: 10
- # 冻结参数
- freeze_params:
- backbone: False,
- rpn: False,
- roi_heads:
- box_head: False,
- box_predictor: False,
- line_head: False,
- line_predictor:
- fc1: False,
- fc2:
- 0: False,
- 2: False,
- 4: False
|
