diff --git a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py index 3aa029827319ea53d29932c326092cae622dc5b7..d13fd38d7ed361f9561da4f005e80d06c4fa01d3 100644 --- a/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py +++ b/downloads-generation/models_class1_pan_unselected/generate_hyperparameters.py @@ -48,9 +48,9 @@ base_hyperparameters = { } grid = [] -for layer_sizes in [[512, 256], [512, 512], [1024, 512]]: +for layer_sizes in [[512, 256], [512, 512], [1024, 512], [1024, 1024]]: for pretrain in [True]: - l1_base = 0.000001 + l1_base = 0.00001 for l1 in [l1_base, l1_base / 10, l1_base / 100, l1_base / 1000, 0.0]: new = deepcopy(base_hyperparameters) new["layer_sizes"] = layer_sizes