config creator.py

import json

# Create a dictionary with the configuration settings
config_dict = {
    "model_type": "custom",
    "hidden_size": 256,
    "num_layers": 3,
    "dropout": 0.2,
    # Add other settings here
}

# Write the dictionary to a JSON file
with open("config.json", "w") as f:
    json.dump(config_dict, f)