EdgePred / config.json
Einae's picture
Update model - Task: default
18b0cfa verified
{
"model_name": "EdgePredMolecularEncoder",
"framework": "torch_molecule",
"date_created": "2025-04-03T11:36:36.524305",
"tasks": {
"default": {
"versions": [],
"current_version": "0.0.2",
"date_added": "2025-04-03T11:36:36.524557",
"last_updated": "2025-04-11T12:00:13.130500",
"num_parameters": 3340505,
"config": {
"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": [
126.87282275248718,
1.018491161010742,
0.771276861858368,
0.7528298636245727,
0.7461581855659485,
0.7435378093070983,
0.74287037915802,
0.741941520778656,
0.7415088617286683,
0.7411704962425232,
0.7408862036476135,
0.7406087256660462,
0.7404774213066101,
0.7402747181930542,
0.7399035019378662,
0.7396295527648926,
0.7393151109848023,
0.7391179640045166,
0.7388970016822815,
0.7387452254447937,
0.7385262452468873,
0.738424241279602,
0.7382177104797363,
0.7381010771903992,
0.7381114503746032,
0.7380318504447937,
0.7379609165878296,
0.7379083976783752,
0.7377160478286743,
0.7378281858139039,
0.7376263963546753,
0.737585072505951,
0.7374617083282471,
0.7375064916801453,
0.7373893736839294,
0.7374962797470093,
0.737427143371582,
0.737349824836731,
0.7372654935455323,
0.7372552788581848,
0.7371928034667968,
0.737183696258545,
0.7372340579147338,
0.7371167575492859,
0.7370882966842651,
0.7370831072769165,
0.7370591225471497,
0.7369692922515869,
0.7370670101928711,
0.7370989799766541,
0.7370519472885132,
0.737019460762024,
0.736972613773346,
0.7369809513626099,
0.7368793008651734,
0.7368705789260864,
0.7368885249137879,
0.7368154951171875,
0.7368716071205139,
0.736730445777893,
0.7368272400054932,
0.7368745569229126,
0.7368906103744507,
0.7367268496856689,
0.7368191294288635,
0.7368267156791687,
0.7367659691696167,
0.7366822110214234,
0.7367576462478638,
0.7367977180175781,
0.7367585034408569,
0.7366750293083191,
0.7366691032180787,
0.7367056628112792,
0.7367450298271179,
0.7366795306510925,
0.7366796090660095,
0.7366272177658081,
0.7365863606376648,
0.7366966720352173,
0.7366529640884399,
0.736596214138031,
0.7366059941368103,
0.7365889531898498,
0.7364908695640564,
0.7366502928085327,
0.7366409742164611,
0.7365723941116333,
0.7365638709983826,
0.7365316697463989,
0.736692658203125,
0.7364824006614685,
0.7365437007179261,
0.7365845192642212,
0.7365730910873413,
0.7364766020507812,
0.7365529715232849,
0.736510915271759,
0.7364802263679504,
0.7364898061218261
],
"device": {
"_type": "unknown",
"repr": "cuda:0"
},
"verbose": false,
"model_name": "EdgePredMolecularEncoder"
},
"metrics": {}
}
},
"last_updated": "2025-04-11T12:00:13.130515"
}