fix message
This commit is contained in:
parent
3a70d09921
commit
7ffd8a51fd
|
@ -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")
|
||||
|
|
Loading…
Reference in a new issue