This commit is contained in:
Markus Friedrich 2020-06-28 17:49:55 +02:00
commit 6b28519e58

View File

@ -17,12 +17,16 @@ 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)