diff --git a/multi_run.py b/multi_run.py index 3b1100b..c7c503a 100644 --- a/multi_run.py +++ b/multi_run.py @@ -17,13 +17,17 @@ if __name__ == '__main__': config = ThisConfig().read_namespace(args) # bias, activation, model, norm, max_epochs pn2 = dict(model_type='PN2', model_use_bias=True, model_use_norm=True, data_batchsize=250) + # p2g = dict(model_type='P2G', model_use_bias=True, model_use_norm=True, data_batchsize=250) # bias, activation, model, norm, max_epochs for arg_dict in [pn2]: for seed in range(2): - arg_dict.update(main_seed=seed) + for poly_as_plane in [True, False]: + for normals_as_cords in [True, False]: + arg_dict.update(main_seed=seed, + normals_as_cords=normals_as_cords, poly_as_plane=poly_as_plane) - config = config.update(arg_dict) + config = config.update(arg_dict) - run_lightning_loop(config) + run_lightning_loop(config)