This commit is contained in:
JiaqiGuo 2019-12-07 11:24:42 +09:00
Родитель 3f8224514c
Коммит 72df5c876f
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -103,11 +103,11 @@ def train(args):
print(tb)
else:
utils.save_checkpoint(model, os.path.join(model_save_path, 'end_model.model'))
json_datas = utils.epoch_acc(model, args.batch_size, val_sql_data, val_table_data,
json_datas, sketch_acc, acc = utils.epoch_acc(model, args.batch_size, val_sql_data, val_table_data,
beam_size=args.beam_size)
acc = utils.eval_acc(json_datas, val_sql_data)
# acc = utils.eval_acc(json_datas, val_sql_data)
print("Sketch Acc: %f, Acc: %f, Beam Acc: %f" % (acc, acc, acc,))
print("Sketch Acc: %f, Acc: %f, Beam Acc: %f" % (sketch_acc, acc, acc,))
if __name__ == '__main__':