Skip to content

Commit 24fd76a

Browse files
committed
Clean up docker invocation script
1 parent c3a8402 commit 24fd76a

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

runDockerTraining.cmd

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,6 @@
1-
set USER=Dzenan
2-
set PROJECT=multic_segment
3-
set BASEDIR=M:/Histo/Work
41
set DATADIR=M:/Histo/TrainSmall
52
set MODELDIR=M:/Histo/Work
63
set CONTAINER=dzenanz/compreps:hpg_training
74
set CUDA_LAUNCH_BLOCKING=1
85
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
106

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)