From df75883615caaaa6c52dfa4a2185c766ab690bd6 Mon Sep 17 00:00:00 2001 From: Kai Han Date: Wed, 7 Dec 2022 21:57:47 +0800 Subject: [PATCH] Update gnn_layers.py --- timm/models/layers/gnn_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/layers/gnn_layers.py b/timm/models/layers/gnn_layers.py index a26220e4..4275cb3c 100755 --- a/timm/models/layers/gnn_layers.py +++ b/timm/models/layers/gnn_layers.py @@ -4,9 +4,9 @@ import numpy as np import torch from torch import nn import torch.nn.functional as F +from timm.models.fx_features import register_notrace_module from .drop import DropPath from .pos_embed import build_sincos2d_pos_embed -from .fx_features import register_notrace_module def pairwise_distance(x, y):