highdeff1 / config creator.py
highdeff's picture
Upload 16 files
2c07569
raw
history blame contribute delete
313 Bytes
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)