| 123456789101112131415161718192021222324252627282930313233343536373839 |
- io:
- logdir: train_results
- datadir: \\192.168.50.222/share/rlq/datasets/0706_
- data_type: tiff
- # 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
- 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
|
