30 lines
847 B
Bash
30 lines
847 B
Bash
|
#!/bin/bash
|
|||
|
|
# 3D检测训练（修复shm问题）
|
|||
|
|
|
|||
|
|
set -e
|
|||
|
|
|
|||
|
|
export PATH=/opt/conda/bin:$PATH
|
|||
|
|
cd /workspace/bevfusion
|
|||
|
|
|
|||
|
|
echo "=========================================="
|
|||
|
|
echo "BEVFusion 3D检测训练"
|
|||
|
|
echo "=========================================="
|
|||
|
|
echo ""
|
|||
|
|
echo "配置: Camera + LiDAR Fusion"
|
|||
|
|
echo "GPU: 8x Tesla V100S"
|
|||
|
|
echo "优化: 减少workers避免shm不足"
|
|||
|
|
echo "预计时间: 20-24小时"
|
|||
|
|
echo ""
|
|||
|
|
echo "=========================================="
|
|||
|
|
echo ""
|
|||
|
|
|
|||
|
|
# 使用命令行参数减少workers_per_gpu
|
|||
|
|
torchpack dist-run -np 8 python tools/train.py \
|
|||
|
|
configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/convfuser.yaml \
|
|||
|
|
--model.encoders.camera.backbone.init_cfg.checkpoint pretrained/swint-nuimages-pretrained.pth \
|
|||
|
|
--load_from pretrained/lidar-only-det.pth \
|
|||
|
|
--data.workers_per_gpu 0
|
|||
|
|
|
|||
|
|
echo ""
|
|||
|
|
echo "训练完成！"
|