Write argmax

atlas_server/src/static/training/predict.job.sh

@@ -26,5 +26,5 @@ export ATLAS_ANNOTATION_DIR=`pwd`/annotations/
 export ATLAS_ANNOTATION_FILE="B${BRAIN}_{slice:04d}_annotations.hdf5"
 
 # Make prediction and run evaluation
-srun --cpus-per-task 4 atlas dl predict `pwd` --verbose --output-spacing ${OUTPUT_SPACING} --overwrite
+srun --cpus-per-task 4 atlas dl predict `pwd` --verbose --output-spacing ${OUTPUT_SPACING} --overwrite --save-argmax
 srun --cpus-per-task 4 atlas dl evaluate `pwd` --verbose --embedded-pred-to-hdf5 --output-spacing ${OUTPUT_SPACING}