Update pytorch_run.py

bigmb · Jun 10, 2019 · 1ae5a3c · 1ae5a3c
1 parent 710771d
commit 1ae5a3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_run.py b/pytorch_run.py
@@ -3,7 +3,7 @@
 import numpy as np
 from PIL import Image
 import glob
-import SimpleITK as sitk
+#import SimpleITK as sitk
 from torch import optim
 import torch.utils.data
 import torch