File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ neuronx_mode = false
1515graviton_mode = false
1616# Please only set it to true if you are preparing a ARM64 related PR
1717# Do remember to revert it back to false before merging any PR (including ARM64 dedicated PR)
18- arm64_mode = false
18+ arm64_mode = true
1919# Please only set it to True if you are preparing a HABANA related PR
2020# Do remember to revert it back to False before merging any PR (including HABANA dedicated PR)
2121habana_mode = false
@@ -122,7 +122,7 @@ use_scheduler = false
122122dlc-pr-base = " "
123123
124124# Standard Framework Training
125- dlc-pr-pytorch-training = " pytorch/training/buildspec-2-8-sm .yml"
125+ dlc-pr-pytorch-training = " pytorch/training/buildspec-arm64-2-7-ec2 .yml"
126126dlc-pr-tensorflow-2-training = " "
127127dlc-pr-autogluon-training = " "
128128
You can’t perform that action at this time.
0 commit comments