steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Training running

Browse Source
This commit is contained in:
Si11ium
2020-06-08 10:23:12 +02:00
parent 27ae8467fc
commit 821b2d1961
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
multi_run.py
View File

@ -1,26 +1,27 @@
import warnings
from _templates.new_project.utils.project_config import Config
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
# Imports
# =============================================================================
from _templates.new_project.main import run_lightning_loop, args
from main import run_lightning_loop
from utils.project_config import ThisConfig
from _parameters import args
if __name__ == '__main__':
# Model Settings
config = Config().read_namespace(args)
config = ThisConfig().read_namespace(args)
# bias, activation, model, norm, max_epochs
cnn_classifier = dict(train_epochs=10, model_use_bias=True, model_use_norm=True, data_batchsize=512)
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 [cnn_classifier]:
for seed in range(5):
for arg_dict in [p2g]:
for seed in range(10):
arg_dict.update(main_seed=seed)
config = config.update(arg_dict)