newron.pytorch.autolog()

newron.pytorch.autolog(log_models=True, disable=False, exclusive=False, disable_for_unsupported_versions=False, silent=False, registered_model_name=None)

Enables (or disables) and configures autologging from pytorch to newron. Logs loss and any other metrics specified in the fit function, and optimizer data as parameters. Model checkpoints are logged as artifacts to a ‘models’ directory.

Parameters

• log_models – If True, trained models are logged as newron model artifacts. If False, trained models are not logged.
• disable – If True, disables the pytorch autologging integration. If False, enables the pytorch autologging integration.
• exclusive – If True, autologged content is not logged to user-created fluent runs. If False, autologged content is logged to the active fluent run, which may be user-created.
• disable_for_unsupported_versions – If True, disable autologging for versions of pytorch that have not been tested against this version of the newron client or are incompatible.
• silent – If True, suppress all event logs and warnings from newron during pytorch autologging. If False, show all events and warnings during pytorch autologging.
• registered_model_name – If given, each time a model is trained, it is registered as a new model version of the registered model with this name. The registered model is created if it does not already exist.