Pconv

chankeh · Dec 24, 2020 · ee9de47 · ee9de47
1 parent 5aab489
commit ee9de47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/2D/model_pconv.py b/2D/model_pconv.py
@@ -27,6 +27,7 @@
 import os
 import time
 import shutil
+import settings
 
 import tensorflow as tf  # conda install -c anaconda tensorflow
 
@@ -83,7 +84,6 @@ def __init__(self, channels_first=settings.CHANNELS_FIRST,
         self.output_path = output_path
         self.inference_filename = inference_filename
 
-        self.batch_size = batch_size
 
         self.metrics = [self.dice_coef, self.soft_dice_coef]