Update README.md
Browse files
README.md
CHANGED
|
@@ -50,8 +50,6 @@ with torch.amp.autocast(
|
|
| 50 |
print(probs)
|
| 51 |
```
|
| 52 |
|
| 53 |
-
Published GPU benchmarks use FP16 mixed precision for model execution. SDM respects the caller's active PyTorch autocast context but does not enable autocasting automatically, so use the context shown above for comparable GPU throughput.
|
| 54 |
-
|
| 55 |
To learn more, visit [structured-data-models](https://github.com/NVIDIA/structured-data-models).
|
| 56 |
|
| 57 |
## License
|
|
|
|
| 50 |
print(probs)
|
| 51 |
```
|
| 52 |
|
|
|
|
|
|
|
| 53 |
To learn more, visit [structured-data-models](https://github.com/NVIDIA/structured-data-models).
|
| 54 |
|
| 55 |
## License
|