27 lines
826 B
Bash
27 lines
826 B
Bash
|
|
#!/bin/bash
|
|||
|
|
|
|||
|
|
echo "========================================"
|
|||
|
|
echo "启动高分辨率BEV分割训练"
|
|||
|
|
echo "========================================"
|
|||
|
|
echo ""
|
|||
|
|
|
|||
|
|
echo "配置:"
|
|||
|
|
echo " - BEV: 180×180 (保持不变)"
|
|||
|
|
echo " - 分割输出: 400×400 (grid_transform上采样)"
|
|||
|
|
echo " - GT标签: 400×400"
|
|||
|
|
echo " - Batch: 2/GPU × 8GPU = 16总batch"
|
|||
|
|
echo " - 基于: epoch_19.pth"
|
|||
|
|
echo ""
|
|||
|
|
|
|||
|
|
export PATH=/opt/conda/bin:$PATH
|
|||
|
|
cd /workspace/bevfusion
|
|||
|
|
|
|||
|
|
/opt/conda/bin/torchpack dist-run \
|
|||
|
|
-np 8 \
|
|||
|
|
/opt/conda/bin/python tools/train.py \
|
|||
|
|
configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/multitask_enhanced_phase1_HIGHRES.yaml \
|
|||
|
|
--model.encoders.camera.backbone.init_cfg.checkpoint pretrained/swint-nuimages-pretrained.pth \
|
|||
|
|
--load_from runs/run-326653dc-74184412/epoch_19.pth \
|
|||
|
|
--data.samples_per_gpu 2
|
|||
|
|
|