import warnings

from ml_lib.utils.config import Config

warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)

# Imports
# =============================================================================

from main import run_lightning_loop, args


if __name__ == '__main__':

    # Model Settings
    config = Config().read_namespace(args)
    # use_bias, activation, model, use_norm, max_epochs, filters
    cnn_classifier = dict(train_epochs=10, model_use_bias=True, model_use_norm=True, model_activation='leaky_relu',
                          model_type='classifier_cnn', model_filters=[16, 32, 64], data_batchsize=512)
    # use_bias, activation, model, use_norm, max_epochs, sr, feature_mixed_dim, filters

    for arg_dict in [cnn_classifier]:
        for seed in range(5):
            arg_dict.update(main_seed=seed)

            config = config.update(arg_dict)

            run_lightning_loop(config)