Artificial Intelligence, Machine Learning, Deep Learning, Image Classification, Heartbeat.
are close to the metrics on the train set. Lastly, let’s save our model in Comet ML and finish our experiment, as follows: # Saving Model in Comet-ML from comet_ml.integration.pytorch import log_model log_model(comet_logger.experiment, my_model, model_name="my_pl_model") # Ending our experiment