Merge pull request #313 from venkatram-dev/fix_box_loss_error

fix_box_loss_error

roboflow/core/version.py

@@ -413,7 +413,7 @@ def live_plot(epochs, mAP, loss, title=""):
                 mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]])
                 loss = np.array(
                     [
-                        (float(epoch["box_loss"]) + float(epoch["class_loss"]) + float(epoch["obj_loss"]))
+                        sum(float(epoch[key]) for key in ["box_loss", "class_loss", "obj_loss"] if key in epoch)
                         for epoch in models["roboflow-train"]["epochs"]
                     ]
                 )