PP-ShiTuV2_rec_CLIP_vit_large.yaml 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. Global:
  2. model: PP-ShiTuV2_rec_CLIP_vit_large
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. dataset_dir: "/paddle/dataset/paddlex/shitu_rec/Inshop_examples"
  5. device: gpu:0,1,2,3
  6. output: "output"
  7. CheckDataset:
  8. convert:
  9. enable: False
  10. src_dataset_type: null
  11. split:
  12. enable: False
  13. train_percent: null
  14. gallery_percent: null
  15. query_percent: null
  16. Train:
  17. num_classes: 159
  18. epochs_iters: 20
  19. batch_size: 32
  20. learning_rate: 0.001
  21. pretrain_weight_path: https://paddleclas.bj.bcebos.com/models/PP-ShiTu/PP-ShiTuV2_rec_CLIP_vit_large_pretrained.pdparams
  22. warmup_steps: 5
  23. resume_path: null
  24. log_interval: 1
  25. eval_interval: 1
  26. save_interval: 1
  27. Evaluate:
  28. weight_path: "output/best_model/best_model.pdparams"
  29. log_interval: 1
  30. Export:
  31. weight_path: https://paddleclas.bj.bcebos.com/models/PP-ShiTu/PP-ShiTuV2_rec_CLIP_vit_large_pretrained.pdparams
  32. Predict:
  33. batch_size: 1
  34. model_dir: "output/best_model/inference"
  35. input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_recognition_001.jpg"
  36. kernel_option:
  37. run_mode: paddle