From 9c9a00f0b7bfd6689b1605eb1599ff719c4af958 Mon Sep 17 00:00:00 2001 From: ahmedhshahin Date: Wed, 20 Dec 2023 17:06:08 +0000 Subject: [PATCH] fix No module named 'torch._six' --- examples/longvit/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/longvit/utils.py b/examples/longvit/utils.py index aa529ecf..2629cad6 100644 --- a/examples/longvit/utils.py +++ b/examples/longvit/utils.py @@ -21,7 +21,7 @@ import torch.distributed as dist import torch.nn as nn import torch.nn.functional as F -from torch._six import inf +from torch import inf # from torchmetrics import Metric from tensorboardX import SummaryWriter