mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-11-28 20:09:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove model name restriction

Browse Source
This commit is contained in:
deeppomf 2018-10-20 00:11:40 -04:00
parent 40d3a50d45
commit a6f60f4784
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/pconv_hybrid_model.py
View File

@ -230,9 +230,9 @@ class PConvUnet(object):
# Load weights into model # Load weights into model
#epoch = 50 #epoch = 50
epoch = int(os.path.basename(filepath).split("_")[0]) # 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" # assert epoch > 0, "Could not parse weight file. Should start with 'X_', with X being the epoch"
self.current_epoch = epoch # self.current_epoch = epoch
self.model.load_weights(filepath) self.model.load_weights(filepath)
def current_weightfile(self): def current_weightfile(self):