| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- batch_size: 1
- iters: 20
- output_op: none
- model:
- type: STFPM
- backbone:
- type: ResNet18
- train_dataset:
- type: Dataset
- num_classes: 1
- dataset_root: /mv_dataset/hazelnut
- transforms:
- - type: Resize
- target_size: [256, 256]
- - type: Normalize
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- mode: train
- val_dataset:
- type: Dataset
- num_classes: 1
- dataset_root: /mv_dataset/hazelnut
- transforms:
- - type: Resize
- target_size: [256, 256]
- - type: Normalize
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- mode: val
- loss:
- types:
- - type: DistillationLoss
- coef: [1]
- optimizer:
- type: SGD
- momentum: 0.9
- weight_decay: 1.0e-4
- lr_scheduler:
- type: PolynomialDecay
- learning_rate: 0.4
- end_lr: 0.4
- power: 0.9
|