Open zstreet87 opened 1 year ago
PyTorch 2.0 work using upstream TOT showed the title error, which is ultimately from this PR https://github.com/pytorch/pytorch/pull/94709.
File apex/amp/_initialize.py ❱ from torch._six import string_classes
apex/amp/_initialize.py
❱ from torch._six import string_classes
Solution is to cherry pick upstream apex for the fix. https://github.com/NVIDIA/apex/commit/6943fd26e04c59327de32592cf5af68be8f5c44e
@jpvillam-amd
zstreet87
commented
1 year ago zstreet87
commented
1 year ago
PyTorch 2.0 work using upstream TOT showed the title error, which is ultimately from this PR https://github.com/pytorch/pytorch/pull/94709.
File
apex/amp/_initialize.py❱ from torch._six import string_classesSolution is to cherry pick upstream apex for the fix. https://github.com/NVIDIA/apex/commit/6943fd26e04c59327de32592cf5af68be8f5c44e