Skip to content

Commit f6fbc0a

Browse files
committed
ADD Documentation to ReadtheDocs for Upgrading torch versions
**Description** **Testing Done** Only documentation updates
1 parent b776850 commit f6fbc0a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

doc/overview.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,11 @@ To train a model by using the SageMaker Python SDK, you:
3030

3131
After you train a model, you can save it, and then serve the model as an endpoint to get real-time inferences or get inferences for an entire dataset by using batch transform.
3232

33+
34+
Important Note:
35+
36+
* When using torch to load Models, it is recommended to use version torch>=2.6.0 and torchvision>=0.17.0
37+
3338
Prepare a Training script
3439
=========================
3540

0 commit comments

Comments
 (0)