Fix for specifying inner CV fold number (#31)

ppdebreuck · Feb 15, 2021 · 14fe3d2 · 14fe3d2
1 parent 07342e9
commit 14fe3d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modnet/models.py b/modnet/models.py
@@ -379,7 +379,7 @@ def fit_preset(
         val_losses = 1e20 * np.ones((len(presets),))
 
         num_nested_folds = 5
-        if isinstance(nested, int):
+        if isinstance(nested, int) and nested != True:
             num_nested_folds = nested
 
         best_model = None