| 1234567891011121314151617181920212223242526272829303132333435363738 |
- io:
- logdir: train_results
- datadir: /data/share/lm/Dataset_all
- # 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: 2
- 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
|
