GitLab

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/308

D37088095 made BC breaking changes in PyTorch, but missed
fixing a few callsites. This is the second diff to fix the affected
tests.

Reviewed By: jerryzh168, wat3rBro

Differential Revision: D37282853

fbshipit-source-id: 945c3628f53cefd2fbc2526630d1567a2b1cee25