Ling-Jun
Ling-Jun
 `save_weights_dir`, `state_dict_path`, `checkpoint_dir` all mean roughly the same, namely, the location/path of where the trained model is stored. Can we reduce the different names to just one? Three different...
If we set custom local dir as `mlflow.yaml`'s `uri`, it will raise an exception.  The culprit is `mlflow.set_tracking_uri()` function.  We can fix it by identifying `uri` being a...