Skip to content
Snippets Groups Projects
Commit a0f29257 authored by Xuan Gu's avatar Xuan Gu
Browse files

Update 3 files

- /MLPerf/training/image_segmentation/pytorch/benchmark.txt
- /MLPerf/training/image_segmentation/pytorch/U-Net3D.def
- /MLPerf/training/image_segmentation/pytorch/build.txt
parent 18096f62
No related branches found
No related tags found
No related merge requests found
Bootstrap: localimage Bootstrap: localimage
From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-devel.sif
%environment %environment
......
...@@ -22,5 +22,5 @@ rm -rf kits19 ...@@ -22,5 +22,5 @@ rm -rf kits19
apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "cd /workspace/unet3d && python3 preprocess_dataset.py --data_dir /raw-data --results_dir /data" apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "cd /workspace/unet3d && python3 preprocess_dataset.py --data_dir /raw-data --results_dir /data"
apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "bash /workspace/unet3d/run_and_time.sh 1" apptainer exec -B ${WORK_DIR}/raw-data:/raw-data -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results $CONTAINER_DIR bash -c "cd /workspace/unet3d && bash run_and_time.sh 1"
apptainer build containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif docker://pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime apptainer build containers/pytorch_1.7.1-cuda11.0-cudnn8-devel.sif docker://pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
MODULE_NAME=U-Net3D MODULE_NAME=U-Net3D
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment