Skip to content

Commit c3a8402

Browse files
committed
Add a batch script to invoke training using a docker image
1 parent efcaf5a commit c3a8402

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

runDockerTraining.cmd

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
set USER=Dzenan
2+
set PROJECT=multic_segment
3+
set BASEDIR=M:/Histo/Work
4+
set DATADIR=M:/Histo/TrainSmall
5+
set MODELDIR=M:/Histo/Work
6+
set CONTAINER=dzenanz/compreps:hpg_training
7+
set CUDA_LAUNCH_BLOCKING=1
8+
docker run -a STDIN -a STDOUT -a STDERR --gpus=all -v M:/Histo/Multi-Compartment-Segmentation/:/exec/ -v %DATADIR%/:/data -v %MODELDIR%/:/model/ --entrypoint python3 %CONTAINER% /exec/multic/segmentationschool/segmentation_school.py --option train --base_dir /model --init_modelfile /model/model_0214999.pth --training_data_dir /data --train_steps 1000 --eval_period 250 --num_workers 0
9+
GOTO :EOF
10+
11+
REM Interactive debugging:
12+
docker run -it --gpus=all -v M:/Histo/Multi-Compartment-Segmentation/:/exec/ -v %DATADIR%/:/data -v %MODELDIR%/:/model/ --entrypoint /bin/bash %CONTAINER%
13+
python3 /exec/multic/segmentationschool/segmentation_school.py --option train --base_dir /model --init_modelfile /model/model_0214999.pth --training_data_dir /data --train_steps 1000 --eval_period 250 --num_workers 0

0 commit comments

Comments
 (0)