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())