Upload pim_module.py

pim_module.py

@@ -231,7 +231,13 @@ class FPN(nn.Module):
                 )
             elif proj_type == "Linear":
                 m = nn.Sequential(
-                    nn.Linear(inputs[node_name].size(-1), inputs[node_name].size(-1)),
+                    in_feat = inputs[node_name]
+                    if isinstance(in_feat, torch.Tensor):
+                        dim = in_feat.size(-1)
+                    else:
+                        raise ValueError(f"Entrée invalide dans FPN: {type(in_feat)} pour node_name={node_name}")
+                    nn.Linear(dim, dim)
+#                    nn.Linear(inputs[node_name].size(-1), inputs[node_name].size(-1)),
                     nn.ReLU(),
                     nn.Linear(inputs[node_name].size(-1), fpn_size),
                 )