Save SoSModel to appropriate location
@@ -44,7 +44,7 @@ if __name__ == '__main__':
rnnmodel = SoSModel(rnn_size=100)
logger.info("Created model")
try:
- rnnmodel.load(os.path.join('models','SoSmodelBW'))
+ rnnmodel.load(os.path.join('models','SoSmodel'))
logger.info("Loaded model")
except:
rnnmodel.initialize()
@@ -74,7 +74,7 @@ if __name__ == '__main__':
logger.info("Validation cost after %s batches: %s",i,cost)
if cost < min_cost:
min_cost = cost
- rnnmodel.save(os.path.join('models','SoSmodelBW'))
+ rnnmodel.save(os.path.join('models','SoSmodel'))
last_save = 0
logger.info('Saving model at validation cost %s', cost)
else:
