We need to import torch.

pcuenq changed pull request status to merged

Sign up or log in to comment