atomgpt / config.json
willengler-uc's picture
Upload 2 files
6190126 verified
{
"id_prop_path": "Out/id_prop.csv",
"prefix": "atomgpt_run",
"model_name": "google/flan-t5-small",
"batch_size": 5,
"max_length": 512,
"num_epochs": 10,
"latent_dim": 1024,
"learning_rate": 0.001,
"test_each_run": true,
"include_struct": false,
"pretrained_path": "",
"seed_val": 42,
"n_train": 650,
"n_val": 81,
"n_test": 81,
"output_dir": "atomgpt_out",
"desc_type": "desc_3",
"convert": true,
"train_ratio": null,
"val_ratio": 0.1,
"test_ratio": 0.1,
"keep_data_order": true
}