Skip to content

Commit f1e9afc

Browse files
committed
format pytorch version
1 parent f82aa82 commit f1e9afc

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

src/sagemaker/fw_utils.py

+13-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,19 @@
7373
"2.6.0",
7474
"2.6.2",
7575
],
76-
"pytorch": ["1.6", "1.6.0", "1.7", "1.7.1", "1.8", "1.8.0", "1.8.1", "1.9", "1.9.0", "1.9.1", "1.10.0"],
76+
"pytorch": [
77+
"1.6",
78+
"1.6.0",
79+
"1.7",
80+
"1.7.1",
81+
"1.8",
82+
"1.8.0",
83+
"1.8.1",
84+
"1.9",
85+
"1.9.0",
86+
"1.9.1",
87+
"1.10.0"
88+
],
7789
}
7890
SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]
7991

0 commit comments

Comments
 (0)