Test

Arno1235 · Mar 20, 2023 · fd8de7e · fd8de7e
1 parent b2b2f33
commit fd8de7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -255,7 +255,7 @@ def pred_image(self, image_in, checkpoint_path, num):
     def dice_eval(self, dataset_validation, checkpoint_path):
         #self.checkpoint = tfe.Checkpoint(model=self.model.net)
         self.checkpoint = tf.train.Checkpoint(model=self.model.net)
-        # self.checkpoint.restore(checkpoint_path) # TEST
+        self.checkpoint.restore(checkpoint_path)
         v_size = dataset_validation.size()
 
         dice = []