diff --git a/downloads-generation/models_class1/generate_hyperparameters.py b/downloads-generation/models_class1/generate_hyperparameters.py index 457678799b5b4101e76ee944c6963cd784e5fc72..fc9ef8bcf72f05802f28eca570a117434ef5f6fa 100644 --- a/downloads-generation/models_class1/generate_hyperparameters.py +++ b/downloads-generation/models_class1/generate_hyperparameters.py @@ -64,7 +64,7 @@ base_hyperparameters = { grid = [] for train_subset in ["all", "quantitative"]: - for minibatch_size in [128, 512]: + for minibatch_size in [128]: for dense_layer_size in [8, 16, 32, 64]: for l1 in [0.0, 0.001, 0.01]: for num_lc in [0, 1, 2]: