pytorch

a9308f9d - py2 fix

py2 fix Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29613 Test Plan: Imported from OSS Differential Revision: D18440212 Pulled By: suo fbshipit-source-id: 4e25a599ea2c649d0b6b4531da5df9b00e7f6380