PET / scripts /launch_retrain.sh
DesonDai's picture
Add files using upload-large-folder tool
212e9d7 verified
Raw
History Blame Contribute Delete
470 Bytes
#!/bin/bash
cd /data/Albus/Brain
CUDA_VISIBLE_DEVICES=1 /data/Albus/miniconda3/bin/python -u scripts/train_pet_foundation_epoch_ckpt.py --backbone medicalnet --encoder-train-scope layer4 --epochs 50 --batch-size 4 --lr 1e-5 --num-workers 2 --output-size 96 96 96 --contrastive-weight 0.2 --regression-weight 1.0 --train-csv metadata/splits/train.csv --val-csv metadata/splits/val.csv --out-dir runs/foundation/remap_epochs > logs/remap_epoch_training.log 2>&1