BEVFusion.yaml 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. Global:
  2. model: BEVFusion
  3. mode: check_dataset
  4. dataset_dir: "/paddle/dataset/paddlex/3d/nuscenes_demo"
  5. device: gpu:0,1,2,3
  6. output: "output"
  7. load_cam_from: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/BEVFusion_camera_pretrained.pdparams"
  8. load_lidar_from: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/BEVFusion_lidar_pretrained.pdparams"
  9. datart_prefix: True
  10. version: "mini"
  11. CheckDataset:
  12. convert:
  13. enable: False
  14. split:
  15. enable: False
  16. Train:
  17. epochs_iters: 2
  18. batch_size: 2
  19. learning_rate: 0.001
  20. warmup_steps: 150
  21. Evaluate:
  22. batch_size: 1
  23. weight_path: output/best_model/model.pdparams
  24. Export:
  25. weight_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/BEVFusion_pretrained.pdparams"
  26. Predict:
  27. batch_size: 1
  28. model_dir: "output/best_model/inference"
  29. input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/det_3d/demo_det_3d/nuscenes_demo_infer.tar"
  30. kernel_option:
  31. run_mode: paddle