From e73a148c6f297e06f1cd3b5599d8c72f528cdf81 Mon Sep 17 00:00:00 2001 From: thibaultdvx Date: Wed, 16 Oct 2024 15:11:27 +0200 Subject: [PATCH] fix numpy issue --- clinicadl/network/cnn/random.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clinicadl/network/cnn/random.py b/clinicadl/network/cnn/random.py index 897a014d1..a75727af0 100644 --- 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)