newron.pytorch.log_model()

newron.keras.log_model(pyt_model, artifact_path)

Enables (or disables) and configures data logging 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 -

• pyt_model - It denotes the tensorflow model developed by the user
• artifact_path - It denotes the directory in artifact_uri to write to

Example - Model parameters remain same as for sklearn