Commit 090857fb authored by Abdallah Atouani's avatar Abdallah Atouani
Browse files

fix tests

parent 30a6e49f
Pipeline #353657 passed with stage
in 1 minute and 15 seconds
......@@ -678,7 +678,7 @@ class CNNSupervisedTrainer_Alexnet:
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.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))
......
......@@ -678,7 +678,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
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.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))
......
......@@ -801,7 +801,7 @@ class CNNSupervisedTrainer_EpisodicMemoryNetwork:
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.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))
......
......@@ -678,7 +678,7 @@ class CNNSupervisedTrainer_LoadNetworkTest:
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.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))
......
......@@ -678,7 +678,7 @@ class CNNSupervisedTrainer_VGG16:
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.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))
......
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