update mlflow train command

README.md

@@ -138,7 +138,7 @@ Training is achieved by running trainScript.py. This scripts accepts a range of
 which runs this script as an MLFLOW project.
 
 ```
-mlflow run [email protected]:Zanna-ResearchTeam/subgrid.git --experiment-name new_models -e train --no-conda -P exp_id=19 -P run_id=afab43b4a6274e29822181c4fdaaf925 -P learning_rate=0/5e-4/15/5e-5/30/5e-6 -P n_epochs=200 -P weight_decay=0.00 -P train_split=0.8 \
+mlflow run . --experiment-name train -e train --env_manager=local -P exp_id=<data_experiment_id> -P run_id=<data_run_id> -P learning_rate=0/5e-4/15/5e-5/30/5e-6 -P n_epochs=200 -P weight_decay=0.00 -P train_split=0.8 \
 -P test_split=0.85 -P model_module_name=models.models1 -P model_cls_name=FullyCNN -P batchsize=4 -P transformation_cls_name=SoftPlusTransform -P submodel=transform3 -P loss_cls_name=HeteroskedasticGaussianLossV2
 ```