New Dataset for per spatial cluster training

This commit is contained in:
Si11ium
2020-06-09 14:08:35 +02:00
parent 821b2d1961
commit 23f3aa878d
10 changed files with 104 additions and 12 deletions

View File

@ -16,11 +16,11 @@ if __name__ == '__main__':
# Model Settings
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)
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 [p2g]:
for arg_dict in [pn2]:
for seed in range(10):
arg_dict.update(main_seed=seed)