From a6f60f47840ee5f8d0cd150a13ef8e8d3c9e18f7 Mon Sep 17 00:00:00 2001 From: deeppomf Date: Sat, 20 Oct 2018 00:11:40 -0400 Subject: [PATCH] remove model name restriction --- libs/pconv_hybrid_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/pconv_hybrid_model.py b/libs/pconv_hybrid_model.py index ebffbe6..1778fbb 100644 --- a/libs/pconv_hybrid_model.py +++ b/libs/pconv_hybrid_model.py @@ -230,9 +230,9 @@ class PConvUnet(object): # Load weights into model #epoch = 50 - epoch = int(os.path.basename(filepath).split("_")[0]) - assert epoch > 0, "Could not parse weight file. Should start with 'X_', with X being the epoch" - self.current_epoch = epoch + # epoch = int(os.path.basename(filepath).split("_")[0]) + # assert epoch > 0, "Could not parse weight file. Should start with 'X_', with X being the epoch" + # self.current_epoch = epoch self.model.load_weights(filepath) def current_weightfile(self):