ddim
ddim copied to clipboard
Replace torch._six.string_classes by str
According to https://github.com/pytorch/pytorch/pull/97863 torch._six has been removed. I propose the following modification to avoid the error "module 'torch' has no attribute '_six'". This solution is also suggested in other projects: https://github.com/NVIDIA/apex/issues/1724.