Remove underscore from PyTorch (viamrobotics#3881)

stevebriskin · May 2, 2024 · efcdadc · efcdadc
1 parent ce5f52e
commit efcdadc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cli/ml_training.go b/cli/ml_training.go
@@ -69,7 +69,7 @@ const (
 	ModelFrameworkUnspecified = ModelFramework("unspecified")
 	ModelFrameworkTFLite      = ModelFramework("tflite")
 	ModelFrameworkTensorFlow  = ModelFramework("tensorflow")
-	ModelFrameworkPyTorch     = ModelFramework("py_torch")
+	ModelFrameworkPyTorch     = ModelFramework("pytorch")
 	ModelFrameworkONNX        = ModelFramework("onnx")
 )