{
  "model_name": "ContextPredMolecularEncoder",
  "framework": "torch_molecule",
  "date_created": "2025-04-03T11:37:21.695061",
  "tasks": {
    "default": {
      "versions": [],
      "current_version": "0.0.2",
      "date_added": "2025-04-03T11:37:21.695422",
      "last_updated": "2025-04-07T23:59:22.182684",
      "num_parameters": 3034505,
      "config": {
        "mode": "cbow",
        "context_size": 2,
        "neg_samples": 1,
        "encoder_type": "gin-virtual",
        "readout": "sum",
        "num_layer": 3,
        "hidden_size": 300,
        "drop_ratio": 0.5,
        "norm_layer": "batch_norm",
        "batch_size": 128,
        "epochs": 100,
        "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": 99,
        "fitting_loss": [
          25119.70502548106,
          760.6808744396129,
          661.786918614911,
          648.9513565311826,
          647.5503657802227,
          636.2335364422514,
          640.5620767673013,
          633.7369939788684,
          638.5661164214413,
          628.304155701334,
          629.660377283634,
          634.3422991236279,
          631.5189795917847,
          630.0378613985285,
          630.1065548194824,
          630.4620400377419,
          628.5100429117429,
          629.0226410011572,
          628.016081832873,
          628.2048041656566,
          629.8034932717832,
          624.7170553375538,
          629.5942777795477,
          628.412315533849,
          624.4476656687913,
          626.6924506958187,
          626.1665147599012,
          626.598504243132,
          626.2429474902234,
          626.2078922465635,
          626.8377691062199,
          624.7524442618854,
          625.0407476412834,
          625.3641053898356,
          624.3577862057338,
          624.8819305188035,
          624.1335654428688,
          625.9394846306957,
          623.6362045229358,
          624.1591971650455,
          623.2961951213135,
          623.7323641039579,
          623.6939235748791,
          627.9627225652067,
          624.6863392951142,
          623.35188047652,
          623.9062817619626,
          622.9244442582212,
          623.9427827856848,
          622.3905574396847,
          625.2202454262779,
          623.2602289731765,
          622.4098801640029,
          623.1099330095303,
          622.9293129382366,
          622.941254615324,
          624.1102434335866,
          623.1152124864,
          621.9502472456517,
          622.7576949660654,
          622.4155421469542,
          621.9102569099529,
          622.1758660029628,
          622.4824817983437,
          621.9634459806116,
          623.7321237895819,
          623.9586134751389,
          622.6675494168926,
          622.2359933907079,
          622.579790855923,
          625.1156122288418,
          622.3997919361062,
          621.7474532128333,
          622.4870538633988,
          621.8197156593587,
          622.1910189502403,
          624.0252417014904,
          621.7136268054549,
          622.7923188193531,
          622.0858478747236,
          623.1039878318097,
          621.5573623469954,
          622.4302452486901,
          621.5379836117174,
          621.6846407197872,
          621.9609127160227,
          621.8188203905878,
          622.282664843673,
          622.3281065376061,
          621.8359578060101,
          621.5448882866868,
          622.2150472616468,
          621.7585053412338,
          621.4820724868572,
          621.4557057843303,
          621.8266726264269,
          622.6807558236859,
          621.6629240409912,
          621.7742273767982,
          622.2860015415138
        ],
        "device": {
          "_type": "unknown",
          "repr": "cuda:0"
        },
        "verbose": false,
        "model_name": "ContextPredMolecularEncoder"
      },
      "metrics": {}
    }
  },
  "last_updated": "2025-04-07T23:59:22.182710"
}