From cf4561ca729c69bd9e5c600ca2dc40b7510c50b2 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Thu, 12 Aug 2021 15:31:02 +0100 Subject: [PATCH] Add FX based FeatureGraphNet capability --- timm/models/fx_features.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/fx_features.py b/timm/models/fx_features.py index 310cc465..3d253444 100644 --- a/timm/models/fx_features.py +++ b/timm/models/fx_features.py @@ -486,4 +486,4 @@ class FeatureGraphNet(nn.Module): self.graph_module = build_feature_graph_net(model, return_nodes) def forward(self, x): - return list(self.graph_module(x).values()) \ No newline at end of file + return list(self.graph_module(x).values())