fix numpy issue

aramis-lab · Oct 16, 2024 · e73a148 · e73a148
1 parent 7b88c7a
commit e73a148
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clinicadl/network/cnn/random.py b/clinicadl/network/cnn/random.py
@@ -208,7 +208,7 @@ def fc_dict_design(n_fcblocks, convolutions, initial_shape, n_classes=2):
         out_channels = last_conv["out_channels"]
         flattened_shape = np.ceil(np.array(initial_shape) / 2**n_conv)
         flattened_shape[0] = out_channels
-        in_features = np.product(flattened_shape)
+        in_features = np.prod(flattened_shape)
 
         # Sample number of FC layers
         ratio = (in_features / n_classes) ** (1 / n_fcblocks)