ContextPred / config.json
Einae's picture
Update model - Task: default
ad5a52d verified
{
"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"
}