feature: support PyTorch 1.7.1 training, inference and data parallel - #2185

Merged

ChoiByungWook merged 9 commits into

eitansela:master

Mar 3, 2021

Commits on Mar 3, 2021

feature: support PyTorch 1.7 training and inference
eitansela
committed
b0a7f19
Update pytorch.json
eitansela
committed
2cc6f07
Merge branch 'master' into master
Dan
authored
869e6e6
Update pytorch.json
eitansela
committed
1d7006c
Merge branch 'master' of https://github.com/eitansela/sagemaker-python-sdk
eitansela
committed
6b326ab
Update pytorch.json
eitansela
committed
5005984
Update conftest.py
eitansela
committed
5f88d4a
update data parallel to support pytorch 1.7.1
ChoiByungWook
committed
caf6580
update data parallel integ test to use local mnist dataset
ChoiByungWook
committed
aa54102