| { | |
| "n_batch": 32, | |
| "n_layer": 24, | |
| "n_embd": 768, | |
| "d_state": 16, | |
| "d_conv": 4, | |
| "expand_factor": 2, | |
| "dt_rank": "auto", | |
| "dt_min": 0.001, | |
| "dt_max": 0.1, | |
| "dt_init": "random", | |
| "dt_scale": 1.0, | |
| "dt_init_floor": 0.0001, | |
| "conv_bias": 1, | |
| "bias": 0, | |
| "max_len": 202, | |
| "d_dropout": 0.1, | |
| "dropout": 0.1, | |
| "lr_start": 3e-5, | |
| "lr_multiplier": 1, | |
| "max_epochs": 500, | |
| "model_path": "../", | |
| "ckpt_filename": "smi-ssed_130.pt", | |
| "data_root": "../../moleculenet/esol", | |
| "dataset_name": "esol", | |
| "measure_name": "measured log solubility in mols per litre", | |
| "checkpoints_folder": "./checkpoints_esol", | |
| "loss_fn": "rmse", | |
| "target_metric": "rmse", | |
| "n_output": 1, | |
| "save_ckpt": 1, | |
| "start_seed": 0, | |
| "train_decoder": 1 | |
| } | |