You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Jul 14, 2025. It is now read-only.
Copy file name to clipboardExpand all lines: README.md
+2-1Lines changed: 2 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -109,6 +109,7 @@ for a given [Learning Approach](#ml-modellearning_approach).
109
109
| -------------------------- | ----------- |
110
110
| ml-model:inference-runtime| Represents a file containing instructions for running a containerized version of the model to generate inferences. See the [Inference/Training Runtimes](#inferencetraining-runtimes) section below for details on related fields. |
111
111
| ml-model:training-runtime| Represents a file containing instructions for running a container to train the model. See the [Inference/Training Runtimes](#inferencetraining-runtimes) section below for details on related fields. |
112
+
| ml-model:checkpoint| Represents a PyTorch checkpoint file that can used to load the model (see [official PyTorch documentation for details](https://pytorch.org/tutorials/beginner/saving_loading_models.html)) |
112
113
113
114
### Inference/Training Runtimes
114
115
@@ -124,7 +125,7 @@ While the Compose file defines nearly all of the parameters required to run the
124
125
directory containing input data should be mounted to the container and to which host directory the output predictions should be written. The Compose
125
126
file MUST define volume mounts for input and output data using the Compose
126
127
[Interpolation syntax](https://github.com/compose-spec/compose-spec/blob/master/spec.md#interpolation). The input data volume MUST be defined by an
127
-
`INPUT_VOLUME` variable and the output data volume MUST be defined by an `OUTPUT_DATA` variable.
128
+
`INPUT_DATA` variable and the output data volume MUST be defined by an `OUTPUT_DATA` variable.
128
129
129
130
For example, the following Compose file snippet would mount the host input directory to `/var/data/input` in the container and would mount the host
130
131
output data directory to `/var/data/output` in the host container. In this contrived example, the script to run the model takes 2 arguments: the
0 commit comments