Update yolo_training.py

0259c0a9 · Bubbliiiing · GitHub · f329b183 · 0259c0a9
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

nets/yolo_training.py nets/yolo_training.py +3 -0

未找到文件。
--- a/nets/yolo_training.py
+++ b/nets/yolo_training.py
@@ -461,3 +461,6 @@ class LossHistory():
        plt.legend(loc="upper right")
        plt.savefig(os.path.join(self.save_path, "epoch_loss_" + str(self.time_str) + ".png"))
+        plt.cla()
+        plt.close("all")