1
0
Fork 0
mirror of synced 2024-05-14 18:02:20 +12:00

fix message

This commit is contained in:
nagadomi 2016-08-30 17:13:52 +09:00
parent 3a70d09921
commit 7ffd8a51fd

View file

@ -398,7 +398,7 @@ local function train()
if score.MSE < best_score then
local test_image = image_loader.load_float(settings.test) -- reload
best_score = score.MSE
print("* Best model is updated")
print("* model has updated")
if settings.save_history then
torch.save(settings.model_file_best, model:clearState(), "ascii")
torch.save(string.format(settings.model_file, epoch, i), model:clearState(), "ascii")