config.json

{
  "model_name": "AttrMaskMolecularEncoder",
  "model_state_dict": "model.pt",
  "hyperparameters": {
    "task": {
      "mask_num": 0,
      "mask_rate": 0.15
    },
    "model": {
      "num_layer": 5,
      "hidden_size": 300,
      "drop_ratio": 0.5,
      "norm_layer": "batch_norm",
      "encoder_type": "gin-virtual",
      "readout": "sum"
    },
    "training": {
      "batch_size": 128,
      "epochs": 500,
      "learning_rate": 0.001,
      "grad_clip_value": null,
      "weight_decay": 0.0
    },
    "scheduler": {
      "use_lr_scheduler": false,
      "scheduler_factor": 0.5,
      "scheduler_patience": 5
    },
    "misc": {
      "verbose": false
    }
  },
  "tasks": {
    "default": {
      "versions": [],
      "current_version": "0.0.9",
      "date_added": "2025-04-01T16:19:31.831623",
      "last_updated": "2025-04-05T20:30:19.259861",
      "num_parameters": 3593223,
      "config": {
        "mask_num": 0,
        "mask_rate": 0.15,
        "encoder_type": "gin-virtual",
        "readout": "sum",
        "num_layer": 5,
        "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": [
          0.18591929196357726,
          0.13938841537189484,
          0.1294001416196823,
          0.12361270745372772,
          0.11978809025382996,
          0.116867989777565,
          0.11459532068514824,
          0.11272422646355629,
          0.11131835706830025,
          0.1100062269756794,
          0.1087346100127697,
          0.10777086010718345,
          0.10673904465007782,
          0.10614725803136825,
          0.10558295664644242,
          0.10478757381248474,
          0.10413479817652703,
          0.10359721025061608,
          0.10310574032974243,
          0.10243229065942765,
          0.10214032355761528,
          0.10153073784136772,
          0.10117247366905212,
          0.10040121994495392,
          0.10045763072872162,
          0.1001096592104435,
          0.09980142726135253,
          0.09960950382614135,
          0.09924821351885796,
          0.09880947463202476,
          0.09863367579007148,
          0.09831961731624603,
          0.09807955391836166,
          0.09773134246873856,
          0.09742031597328187,
          0.0974303740272522,
          0.09731282098650933,
          0.09694188970613479,
          0.0964515350587368,
          0.09635353285813332,
          0.09618644301748276,
          0.09597114046573639,
          0.09585660565972329,
          0.09556689257502556,
          0.09563549737405777,
          0.09541711146211625,
          0.09517620991897582,
          0.09511827756381035,
          0.09484247889876365,
          0.09472261615467072,
          0.09454333018231392,
          0.09435330780529975,
          0.09441806952166558,
          0.09400904282855987,
          0.09444060890650749,
          0.0942746956243515,
          0.09391000089621544,
          0.09372787975621223,
          0.0934634834215641,
          0.09327390489435196,
          0.09316657194209099,
          0.0932826163892746,
          0.09327241311883927,
          0.0927676224000454,
          0.09297605008292198,
          0.09282015754151345,
          0.0924869108042717,
          0.09267187008070946,
          0.0925426818716526,
          0.09247001563620567,
          0.09193027105665207,
          0.09220727930903434,
          0.0921467784562111,
          0.09194621335172654,
          0.09201558442044258,
          0.09185899970459938,
          0.09180195833158493,
          0.09205588206982612,
          0.09162687872004509,
          0.09155376754045487,
          0.09136439908194542,
          0.09138098003339767,
          0.09122328493094445,
          0.09108610749435425,
          0.09108968795228005,
          0.09092820961308479,
          0.09104146724653243,
          0.09067323441195488,
          0.09076569167804718,
          0.09075475749135017,
          0.09068126047921181,
          0.09062208874893188,
          0.09052020057320595,
          0.09039648258924485,
          0.09024686970400811,
          0.09019092855763436,
          0.09018145706272125,
          0.09012162613868713,
          0.090243535861969,
          0.08996200981736183
        ],
        "device": {
          "_type": "unknown",
          "repr": "cuda:0"
        },
        "verbose": false,
        "model_name": "AttrMaskMolecularEncoder"
      },
      "metrics": {}
    }
  },
  "last_updated": "2025-04-05T20:30:19.259878"
}