Update utils_fit.py

bubbliiiing · Jan 3, 2022 · a73c7be · a73c7be
1 parent 6484562
commit a73c7be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/utils_fit.py b/utils/utils_fit.py
@@ -114,7 +114,7 @@ def fit_one_epoch(model_train, model, loss_history, loss, optimizer, epoch, epoc
         _, _, accuracy, _, _, _, _ = evaluate(distances,labels)
         print('LFW_Accuracy: %2.5f+-%2.5f' % (np.mean(accuracy), np.std(accuracy)))
 
-    loss_history.append_loss(np.mean(total_accuracy), (total_triple_loss + total_CE_loss) / epoch_step, (val_total_triple_loss + val_total_CE_loss) / epoch_step_val)
+    loss_history.append_loss(total_accuracy / epoch_step, (total_triple_loss + total_CE_loss) / epoch_step, (val_total_triple_loss + val_total_CE_loss) / epoch_step_val)
 
     print('Epoch:'+ str(epoch+1) + '/' + str(Epoch))
     print('Total Loss: %.4f' % ((total_triple_loss + total_CE_loss) / epoch_step))