34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import warnings
|
|
|
|
warnings.filterwarnings('ignore', category=FutureWarning)
|
|
warnings.filterwarnings('ignore', category=UserWarning)
|
|
|
|
# Imports
|
|
# =============================================================================
|
|
|
|
from main import run_lightning_loop
|
|
from utils.project_config import ThisConfig
|
|
from _parameters import args
|
|
|
|
|
|
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)
|
|
# bias, activation, model, norm, max_epochs
|
|
|
|
for arg_dict in [pn2]:
|
|
for seed in range(2):
|
|
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)
|
|
|
|
run_lightning_loop(config)
|