Back out part of "Fix NERPredictor for zero initialization"
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17482
Reviewed By: david-y-lam
Differential Revision: D14216135
fbshipit-source-id: 2ef4cb5dea74fc5c68e9b8cb43fcb180f219cb32