all debug and train running
This commit is contained in:
35
multi_run.py
Normal file
35
multi_run.py
Normal file
@ -0,0 +1,35 @@
|
||||
import warnings
|
||||
|
||||
from lib.utils.config import Config
|
||||
|
||||
warnings.filterwarnings('ignore', category=FutureWarning)
|
||||
warnings.filterwarnings('ignore', category=UserWarning)
|
||||
|
||||
# Imports
|
||||
# =============================================================================
|
||||
from pathlib import Path
|
||||
import os
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
# Model Settings
|
||||
warnings.filterwarnings('ignore', category=FutureWarning)
|
||||
# use_bias, activation, model, use_norm, max_epochs, filters
|
||||
cnn_classifier = [True, 'leaky_relu', 'classifier_cnn', False, 2, [16, 32, 64]]
|
||||
# use_bias, activation, model, use_norm, max_epochs, sr, lat_dim, filters
|
||||
|
||||
# Data Settings
|
||||
data_shortcodes = ['mid', 'mid_5']
|
||||
|
||||
# Iteration over
|
||||
for data_shortcode in data_shortcodes:
|
||||
for use_bias, activation, model, use_norm, max_epochs, filters in [cnn_classifier]:
|
||||
for seed in range(5):
|
||||
arg_dict = dict(main_seed=seed, train_max_epochs=max_epochs,
|
||||
model_use_bias=use_bias, model_use_norm=use_norm,
|
||||
model_activation=activation, model_type=model,
|
||||
model_filters=filters,
|
||||
data_batch_size=512)
|
||||
|
||||
os.system(f'/home/steffen/envs/traj_gen/bin/python main.py {arg_dict}')
|
Reference in New Issue
Block a user