rssr.yaml 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. # Basic configurations of RSSR dataset
  2. seed: 1024
  3. datasets:
  4. train: !Node
  5. type: ResDataset
  6. args:
  7. data_dir: ./test_tipc/data/rssr/
  8. file_list: ./test_tipc/data/rssr/train.txt
  9. num_workers: 0
  10. shuffle: True
  11. sr_factor: 4
  12. eval: !Node
  13. type: ResDataset
  14. args:
  15. data_dir: ./test_tipc/data/rssr/
  16. file_list: ./test_tipc/data/rssr/val.txt
  17. num_workers: 0
  18. shuffle: False
  19. sr_factor: 4
  20. transforms:
  21. train:
  22. - !Node
  23. type: DecodeImg
  24. - !Node
  25. type: RandomCrop
  26. args:
  27. crop_size: 32
  28. - !Node
  29. type: RandomHorizontalFlip
  30. args:
  31. prob: 0.5
  32. - !Node
  33. type: RandomVerticalFlip
  34. args:
  35. prob: 0.5
  36. - !Node
  37. type: Normalize
  38. args:
  39. mean: [0.0, 0.0, 0.0]
  40. std: [1.0, 1.0, 1.0]
  41. eval:
  42. - !Node
  43. type: DecodeImg
  44. - !Node
  45. type: Resize
  46. args:
  47. target_size: 256
  48. - !Node
  49. type: Normalize
  50. args:
  51. mean: [0.0, 0.0, 0.0]
  52. std: [1.0, 1.0, 1.0]
  53. download_on: False
  54. download_url: https://paddlers.bj.bcebos.com/datasets/rssr.zip
  55. download_path: ./test_tipc/data/
  56. num_epochs: 10
  57. train_batch_size: 4
  58. save_interval_epochs: 10
  59. log_interval_steps: 10
  60. save_dir: ./test_tipc/output/res/
  61. learning_rate: 0.0005
  62. early_stop: False
  63. early_stop_patience: 5
  64. use_vdl: False
  65. resume_checkpoint: ''