Add missing import (#72840)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72840

Reviewed By: H-Huang

Differential Revision: D34242612

Pulled By: albanD

fbshipit-source-id: 3dd34de96dbf1ae8f3c3ea45888d211d95862c49
(cherry picked from commit d2650ffa75)
This commit is contained in:
Sergii Dymchenko 2022-02-15 11:36:21 -08:00 committed by PyTorch MergeBot
parent da07d1cda2
commit 313557a613

View File

@ -1,6 +1,7 @@
import itertools
import math
from copy import deepcopy
import warnings
import torch
from torch.nn import Module