Shortcuts

MLFlowLogger

torchrl.record.loggers.mlflow.MLFlowLogger(exp_name: str, tracking_uri: str, tags: Optional[Dict[str, Any]] = None, **kwargs) None[source]

Wrapper for the mlflow logger.

Parameters:
  • exp_name (str) – The name of the experiment.

  • tracking_uri (str) – A tracking URI to a datastore that supports MLFlow or a local directory.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources