From 5fb24a4e0becf569fd39c208428b7d685d70e396 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniel=20Garc=C3=ADa?= <50992677+danigarci1@users.noreply.github.com> Date: Tue, 29 Aug 2023 17:54:28 +0200 Subject: [PATCH] Replaces container_abcs with collections.abc --- model/backbones/vit_pytorch.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/model/backbones/vit_pytorch.py b/model/backbones/vit_pytorch.py index 66cf849..16e928a 100644 --- a/model/backbones/vit_pytorch.py +++ b/model/backbones/vit_pytorch.py @@ -27,13 +27,12 @@ import torch import torch.nn as nn import torch.nn.functional as F -from torch._six import container_abcs - +import collections.abc # From PyTorch internals def _ntuple(n): def parse(x): - if isinstance(x, container_abcs.Iterable): + if isinstance(x, collections.abc.Iterable): return x return tuple(repeat(x, n)) return parse