Fix retry on connect failure decorator (#68600)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/68541 by checking string contains instead of exact eror
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68600
Reviewed By: dagitses, H-Huang
Differential Revision: D32535592
Pulled By: rohan-varma
fbshipit-source-id: 864c3e3c6831f2351c2949b2348af4f48a308522