Commit a083afec authored by Abdallah Atouani's avatar Abdallah Atouani
Browse files

fix failing test

parent 0bd1237c
Pipeline #353925 failed with stage
in 11 seconds
......@@ -674,8 +674,13 @@ class CNNSupervisedTrainer_mnist_mnistClassifier_net:
global_loss_test /= (test_batches * single_pu_batch_size)
test_metric_name = metric.get()[0]
test_metric_score = metric.get()[1]
metric_file = open(self._net_creator._model_dir_ + 'metric.txt', 'w')
metric_file.write(test_metric_name + " " + str(test_metric_score))
metric_file.close()
logging.info("Epoch[%d] Train metric: %f, Test metric: %f, Train loss: %f, Test loss: %f" % (epoch, train_metric_score, test_metric_score, global_loss_train, global_loss_test))
if (epoch+1) % checkpoint_period == 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment