STFPM.yaml 801 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. batch_size: 1
  2. iters: 20
  3. output_op: none
  4. model:
  5. type: STFPM
  6. backbone:
  7. type: ResNet18
  8. train_dataset:
  9. type: Dataset
  10. num_classes: 1
  11. dataset_root: /mv_dataset/hazelnut
  12. transforms:
  13. - type: Resize
  14. target_size: [256, 256]
  15. - type: Normalize
  16. mean: [0.485, 0.456, 0.406]
  17. std: [0.229, 0.224, 0.225]
  18. mode: train
  19. val_dataset:
  20. type: Dataset
  21. num_classes: 1
  22. dataset_root: /mv_dataset/hazelnut
  23. transforms:
  24. - type: Resize
  25. target_size: [256, 256]
  26. - type: Normalize
  27. mean: [0.485, 0.456, 0.406]
  28. std: [0.229, 0.224, 0.225]
  29. mode: val
  30. loss:
  31. types:
  32. - type: DistillationLoss
  33. coef: [1]
  34. optimizer:
  35. type: SGD
  36. momentum: 0.9
  37. weight_decay: 1.0e-4
  38. lr_scheduler:
  39. type: PolynomialDecay
  40. learning_rate: 0.4
  41. end_lr: 0.4
  42. power: 0.9