File size: 1,802 Bytes
72adf38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
{
"version": "NA",
"dataset": "custom",
"target": "target",
"atom_features": "cgcnn",
"neighbor_strategy": "k-nearest",
"id_tag": "id",
"random_seed": 123,
"classification_threshold": null,
"n_val": null,
"n_test": null,
"n_train": null,
"train_ratio": 0.8,
"val_ratio": 0.1,
"test_ratio": 0.1,
"target_multiplication_factor": null,
"epochs": 300,
"batch_size": 64,
"weight_decay": 1e-05,
"learning_rate": 0.001,
"filename": "sample",
"warmup_steps": 2000,
"criterion": "mse",
"optimizer": "adamw",
"scheduler": "onecycle",
"pin_memory": false,
"save_dataloader": false,
"write_checkpoint": true,
"write_predictions": true,
"store_outputs": true,
"progress": true,
"log_tensorboard": false,
"standard_scalar_and_pca": false,
"use_canonize": true,
"num_workers": 4,
"cutoff": 6.0,
"max_neighbors": 25,
"keep_data_order": false,
"distributed": false,
"n_early_stopping": null,
"output_dir": "./mp_bulk_modulus",
"matrix_input": false,
"pyg_input": true,
"use_lattice": true,
"use_angle": false,
"mean_train": 101.28746090254813,
"std_train": 73.24307420840046,
"model": {
"name": "iComformer",
"conv_layers": 4,
"edge_layers": 1,
"atom_input_features": 92,
"edge_features": 256,
"triplet_input_features": 256,
"node_features": 256,
"fc_layers": 1,
"fc_features": 256,
"output_features": 1,
"node_layer_head": 1,
"edge_layer_head": 1,
"nn_based": false,
"link": "identity",
"zero_inflated": false,
"use_angle": false,
"angle_lattice": false,
"classification": false
}
} |