pytorch

c47c6cf8 - Try moving assignment to name earlier.

Try moving assignment to name earlier. The assignment invalidates the nn.Module dictionary guards, which seems to lead to perf regressions on hf_Bert.