config.json

{
  "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
}