{
  "model_name": "AttrMaskMolecularEncoder",
  "framework": "torch_molecule",
  "date_created": "2025-07-28T00:54:19.004999",
  "tasks": {
    "default": {
      "versions": [],
      "current_version": "0.0.10",
      "date_added": "2025-07-28T00:54:19.007002",
      "last_updated": "2025-08-07T01:05:27.840307",
      "num_parameters": 3832927,
      "config": {
        "mask_rate": 0.15,
        "lw_rec": 0.5,
        "encoder_type": "gin-virtual",
        "readout": "sum",
        "num_layer": 5,
        "hidden_size": 300,
        "drop_ratio": 0.5,
        "norm_layer": "batch_norm",
        "batch_size": 32,
        "epochs": 10,
        "learning_rate": 0.001,
        "weight_decay": 0.0,
        "grad_clip_value": null,
        "use_lr_scheduler": false,
        "scheduler_factor": 0.5,
        "scheduler_patience": 5,
        "fitting_epoch": 9,
        "fitting_loss": [
          102.70604839920998,
          80.42498134076595,
          73.70141759514809,
          67.83666904270649,
          64.89766776561737,
          61.69467528164387,
          59.701326459646225,
          57.14562127739191,
          54.80569655448198,
          53.3315319865942
        ],
        "device": {
          "_type": "unknown",
          "repr": "cuda:0"
        },
        "verbose": false,
        "model_name": "MoamaMolecularEncoder"
      },
      "metrics": {}
    }
  },
  "last_updated": "2025-08-07T01:05:27.840314"
}