model_containers

model_containers.py

model_containers = {}
for model_name in config.model_names:
    loaded_model = get_loaded_model(config, model_name)
    init_stats = init_model_stats(id2label)
    f1_metric, acc_metric = init_metrics()
    model_containers[model_name] = {
        'loaded_model': loaded_model,
        'stats': init_stats,
        'metrics':{
            'f1': f1_metric,
            'acc': acc_metric
        }
    }