Skip to main content

Model Artifact Description

This document describes the artifact that is produced by training with protege.

Files​

  • weights.pt: The weights of the trained model.
  • run_config.json: A dump of the user supplied arguments to training.
  • metrics.json: Contains the training metrics for the run.
  • augmentations.json: Contains the augmentations for training/inference supplied by the user.
  • hyperparams.json: Contains the hyperparameters for the training run.