8d6f6fb8fab89d858a392ae59960f1e894bdefba,train.py,,train,#Any#Any#Any#Any#,68

Before Change

	model.end_train_step()
	model.save_model(trial=trial, output_log=True)

	if FLAGS.evaluate_dataset == "":
		mse = model.evaluate_test_batch()
		model.print_status(mse)
	else:
		if FLAGS.evaluate_dataset == "all":
			test_list = ["set5", "set14", "bsd100"]
		else:
			test_list = [FLAGS.evaluate_dataset]

		for test_data in test_list:
			test(model, test_data)


def test(model, test_data):
	test_filenames = util.get_files_in_directory(FLAGS.data_dir + "/" + test_data)
	total_psnr = total_mse = 0

After Change

			// training epoch finished
			model.epochs_completed += 1
			mse, psnr = model.evaluate(test_filenames)
			model.print_status(mse, log=model_updated)
			model.log_to_tensorboard(test_filenames[0], psnr, save_meta_data=model_updated)

			model_updated = model.update_epoch_and_lr()

Instances