fix_box_loss_error

venkatram-dev · venkatram-dev · commit f145e9d33d52 · 2024-08-20T12:46:58.000-07:00
diff --git 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"]
                     ]
                 )

Original file line number	Diff line number	Diff line change
`@@ -413,7 +413,7 @@ def live_plot(epochs, mAP, loss, title=""):`
`413`	`413`	`mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]])`
`414`	`414`	`loss = np.array(`
`415`	`415`	`[`
`416`		`- (float(epoch["box_loss"]) + float(epoch["class_loss"]) + float(epoch["obj_loss"]))`
	`416`	`+ sum(float(epoch[key]) for key in ['box_loss', 'class_loss', 'obj_loss'] if key in epoch)`
`417`	`417`	`for epoch in models["roboflow-train"]["epochs"]`
`418`	`418`	`]`
`419`	`419`	`)`