setting.json

import json

config = {
    "input_dim": 13,
    "model_dim": 64,
    "num_heads": 8,
    "num_layers": 2,
    "output_dim": 1,
    "model_type": "custom_transformer"
}

config_save_path = r'setting.json'
with open(config_save_path, 'w') as f:
    json.dump(config, f)