SeaFormer_large.yaml 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. batch_size: 8
  2. iters: 80000
  3. model:
  4. type: SeaFormerSeg
  5. backbone:
  6. type: SeaFormer_large
  7. pretrained: https://bj.bcebos.com/paddleseg/dygraph/ade20k/backbone/seaformer_large_imagenet_pretrained.pdparams
  8. head_channels: 192
  9. embed_dims: [128, 160, 192]
  10. in_index: [0, 1, 2, 3]
  11. train_dataset:
  12. type: Dataset
  13. dataset_root: data/Cityscapes/
  14. train_path: datasets/Cityscapes/train.txt
  15. num_classes: 19
  16. transforms:
  17. - type: ResizeStepScaling
  18. min_scale_factor: 0.5
  19. max_scale_factor: 2.0
  20. scale_step_size: 0.25
  21. - type: RandomPaddingCrop
  22. crop_size: [512, 512]
  23. - type: RandomHorizontalFlip
  24. - type: RandomDistort
  25. brightness_range: 0.4
  26. contrast_range: 0.4
  27. saturation_range: 0.4
  28. - type: Normalize
  29. mode: train
  30. val_dataset:
  31. type: Dataset
  32. dataset_root: datasets/Cityscapes
  33. val_path: datasets/Cityscapes/val.txt
  34. num_classes: 19
  35. transforms:
  36. - type: Normalize
  37. mode: val
  38. optimizer:
  39. type: AdamW
  40. beta1: 0.9
  41. beta2: 0.999
  42. weight_decay: 0.01
  43. custom_cfg:
  44. - name: pos_emb
  45. weight_decay_mult: 0.0
  46. - name: head
  47. lr_mult: 10.0
  48. - name: norm
  49. weight_decay_mult: 0.0
  50. lr_scheduler:
  51. type: PolynomialDecay
  52. learning_rate: 0.00025
  53. power: 1.0
  54. warmup_iters: 1500
  55. warmup_start_lr: 1.0e-6
  56. end_lr: 0
  57. loss:
  58. types:
  59. - type: CrossEntropyLoss
  60. coef: [1]