Train with modelType parameter

Author: digaobarbosa

roboflow/core/version.py

@@ -290,12 +290,13 @@ def export(self, model_format=None):
             except json.JSONDecodeError:
                 response.raise_for_status()
 
-    def train(self, speed=None, checkpoint=None, plot_in_notebook=False) -> InferenceModel:
+    def train(self, speed=None, modelType=None, checkpoint=None, plot_in_notebook=False) -> InferenceModel:
         """
         Ask the Roboflow API to train a previously exported version's dataset.
 
         Args:
             speed: Whether to train quickly or accurately. Note: accurate training is a paid feature. Default speed is `fast`.
+            modelType: The type of model to train. Default depends on kind of project. It takes precedence over speed. You can check the list of model ids by sending a invalid parameter in this argument.
             checkpoint: A string representing the checkpoint to use while training
             plot: Whether to plot the training results. Default is `False`.
 
@@ -328,12 +329,15 @@ def train(self, speed=None, checkpoint=None, plot_in_notebook=False) -> Inferenc
         url = f"{API_URL}/{workspace}/{project}/{self.version}/train"
 
         data = {}
-        if speed:
-            data["speed"] = speed
+        if modelType:
+            data["modelType"] = modelType
 
         if checkpoint:
             data["checkpoint"] = checkpoint
 
+        if speed:
+            data["speed"] = speed
+
         write_line("Reaching out to Roboflow to start training...")
 
         response = requests.post(url, json=data, params={"api_key": self.__api_key})

tests/manual/debugme.py

@@ -51,7 +51,7 @@ def run_cli():
 
 def run_api_train():
     rf = Roboflow()
-    project = rf.workspace("meh3").project("mosquitobao")
+    project = rf.workspace("model-evaluation-workspace").project("donut-2-lcfx0")
     # version_number = project.generate_version(
     #     settings={
     #         "augmentation": {
@@ -63,7 +63,7 @@ def run_api_train():
     #         },
     #     }
     # )
-    version_number = "61"
+    version_number = "4"
     print(version_number)
     version = project.version(version_number)
     model = version.train(