diff --git a/roboflow/core/version.py b/roboflow/core/version.py
index 42c24dc..60fe4a4 100644
--- a/roboflow/core/version.py
+++ b/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"]
                     ]
                 )