22 lines
764 B
Bash
Executable File
22 lines
764 B
Bash
Executable File
#!/bin/bash
|
||
set -e
|
||
|
||
export PATH=/opt/conda/bin:$PATH
|
||
export LD_LIBRARY_PATH=/opt/conda/lib:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
|
||
cd /workspace/bevfusion
|
||
|
||
echo "Phase 4A: BEV 2x训练 - 启动中..."
|
||
|
||
# 直接使用python而不是torchpack
|
||
nohup python -m torch.distributed.launch --nproc_per_node=6 --master_port=29500 \
|
||
tools/train.py \
|
||
configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/multitask_BEV2X_phase4a.yaml \
|
||
--launcher pytorch \
|
||
--model.encoders.camera.backbone.init_cfg.checkpoint=pretrained/swint-nuimages-pretrained.pth \
|
||
--load_from runs/enhanced_from_epoch19/epoch_23.pth \
|
||
--run-dir runs/phase4a_bev2x \
|
||
> phase4a_bev2x.log 2>&1 &
|
||
|
||
echo "✅ 训练已启动，PID: $!"
|
||
echo "日志: phase4a_bev2x.log"
|