Skip to content

Commit

Permalink
doc
Browse files Browse the repository at this point in the history
  • Loading branch information
Frank Guibert committed Jul 4, 2024
1 parent a92cd64 commit 9d3f667
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,10 @@
- unetr++
- [NamedTensors](#namedtensors)
- [Installation](#installation)
- [Usage](#usage)
- [Instanciate a model](#instanciate-a-model)
- [Export to onnx](#export-to-onnx)
- [NamedTensors](#namedtensors-example)
- [Running tests](#tests)

# Neural Network Architectures
Expand Down Expand Up @@ -101,7 +105,7 @@ output_tensor = onnx_load_and_infer("model.onnx", input_tensor)

Check the code of [onnx_load_and_infer](mfai/torch/__init__.py#L35) if you wouls like to load the model once and make multiple inferences.

## NamedTensors
## NamedTensors example

We use **NamedTensor** instances to keep the link between our torch tensors and our physical/weather feature names (for plotting, for specific losses weights on given features, ...).

Expand Down

0 comments on commit 9d3f667

Please sign in to comment.