Don't close MLflow run context when training model
Don't close MLflow run context when training model: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Is your feature request related to a problem? Please describe. Currently I am using MLflow projects to run a python script that trains a model using ludwig. I would like to…
The task
Is your feature request related to a problem? Please describe. Currently I am using MLflow projects to run a python script that trains a model using ludwig. I would like to set the experiment name and model name to the one I pass via CLI parameters. At the same time the model…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.