Update model - Task: default

MoamaMolecularEncoder.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00f2ae94376545adbc81a870e2a0d5f8cdf755db7ae3f517ccaba78d563b3039
-size 15498975
+oid sha256:77602a5b913d1551f52934ddbe68af2820cb211787fe287a622a1dca05e45eab
+size 15467359

README.md

@@ -1,32 +1,31 @@
 ---
 tags:
 - torch_molecule
-- molecular-property-prediction
 library_name: torch_molecule
 ---
 
-# AttrMaskMolecularEncoder Model
+# MoamaMolecularEncoder Model
 
 ## Model Description
-- **Model Type**: AttrMaskMolecularEncoder
+- **Model Type**: MoamaMolecularEncoder
 - **Framework**: torch_molecule
-- **Last Updated**: 2025-07-31
+- **Last Updated**: 2025-08-07
 
 ## Task Summary
 | Task | Version | Last Updated | Parameters | Metrics |
 |------|---------|--------------|------------|----------|
-| default | 0.0.8 | 2025-07-31 | 3,594,425 |  |
+| default | 0.0.9 | 2025-08-07 | 3,832,927 |  |
 
 
 ## Usage
 
 ```python
-from torch_molecule import AttrMaskMolecularEncoder
+from torch_molecule import MoamaMolecularEncoder
 
 # Load model for specific task
-model = AttrMaskMolecularEncoder()
+model = MoamaMolecularEncoder()
 model.load(
-    "local_model_dir/AttrMaskMolecularEncoder.pt",
+    "local_model_dir/MoamaMolecularEncoder.pt",
     repo="Do1Yun/TorchMoleculeEncoderRepo"
 )
 
@@ -42,34 +41,34 @@ model.load(
 
 
 ### default Task
-- **Current Version**: 0.0.8
-- **Last Updated**: 2025-07-31
-- **Parameters**: 3,594,425
+- **Current Version**: 0.0.9
+- **Last Updated**: 2025-08-07
+- **Parameters**: 3,832,927
 - **Configuration**:
 ```python
 {
-  "mask_num": 0,
   "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": 128,
-  "epochs": 500,
+  "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": 499,
+  "fitting_epoch": 9,
   "device": {
     "_type": "unknown",
     "repr": "cuda:0"
   },
   "verbose": false,
-  "model_name": "AttrMaskMolecularEncoder"
+  "model_name": "MoamaMolecularEncoder"
 }
 ```

config.json

@@ -5,539 +5,49 @@
   "tasks": {
     "default": {
       "versions": [],
-      "current_version": "0.0.8",
+      "current_version": "0.0.9",
       "date_added": "2025-07-28T00:54:19.007002",
-      "last_updated": "2025-07-31T11:23:16.269022",
-      "num_parameters": 3594425,
+      "last_updated": "2025-08-07T01:00:46.754264",
+      "num_parameters": 3832927,
       "config": {
-        "mask_num": 0,
         "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": 128,
-        "epochs": 500,
+        "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": 499,
+        "fitting_epoch": 9,
         "fitting_loss": [
-          4.970042705535889,
-          4.398396015167236,
-          4.528616428375244,
-          3.597484588623047,
-          3.4864134788513184,
-          3.117708444595337,
-          2.6083717346191406,
-          2.4843525886535645,
-          2.0031826496124268,
-          1.7559397220611572,
-          1.4883317947387695,
-          1.991655945777893,
-          1.823971152305603,
-          0.9272733926773071,
-          0.8319627046585083,
-          1.4395474195480347,
-          0.9929925203323364,
-          0.6721078157424927,
-          0.8488162755966187,
-          0.6403642892837524,
-          1.1394836902618408,
-          0.5786928534507751,
-          0.6110153198242188,
-          0.5016398429870605,
-          0.9808347821235657,
-          0.8447147607803345,
-          0.698067307472229,
-          0.509992241859436,
-          0.18249700963497162,
-          0.8892008066177368,
-          0.578630805015564,
-          0.7399061918258667,
-          0.5619465112686157,
-          0.5687018036842346,
-          0.3927299380302429,
-          0.6857832074165344,
-          0.6831801533699036,
-          0.588654637336731,
-          0.6436779499053955,
-          0.4936468005180359,
-          0.5194535255432129,
-          0.4830659031867981,
-          0.6240726113319397,
-          0.5283722877502441,
-          0.4303649961948395,
-          0.7247880697250366,
-          0.46519726514816284,
-          0.3725079298019409,
-          0.6541604995727539,
-          0.9007629156112671,
-          0.508566677570343,
-          0.5189288854598999,
-          0.19410741329193115,
-          0.2392125427722931,
-          0.9587053060531616,
-          0.5543688535690308,
-          0.38239896297454834,
-          0.1782798171043396,
-          0.6457817554473877,
-          0.5448626279830933,
-          0.4656515121459961,
-          0.5105409026145935,
-          0.7516538500785828,
-          0.7143715023994446,
-          0.2984829545021057,
-          0.6984550356864929,
-          0.6106905937194824,
-          0.40352025628089905,
-          0.5442360639572144,
-          0.7950905561447144,
-          0.36293932795524597,
-          0.4497614800930023,
-          0.3197915554046631,
-          0.32281872630119324,
-          0.3995891511440277,
-          0.831814169883728,
-          0.398742139339447,
-          0.5718639492988586,
-          0.47324198484420776,
-          0.4450412690639496,
-          0.4061422348022461,
-          0.565984308719635,
-          0.5918912887573242,
-          0.5503686666488647,
-          0.6086817979812622,
-          1.0489857196807861,
-          0.4702257513999939,
-          0.41339054703712463,
-          0.8361985087394714,
-          0.4894932806491852,
-          1.1718226671218872,
-          0.4371999204158783,
-          0.4521377980709076,
-          0.5549439191818237,
-          0.2188672125339508,
-          0.22650480270385742,
-          0.9554474949836731,
-          0.32810860872268677,
-          0.3269241750240326,
-          0.36918941140174866,
-          0.5488607883453369,
-          0.33190685510635376,
-          0.40350133180618286,
-          0.09132788330316544,
-          0.6231361627578735,
-          0.7675579786300659,
-          0.6165639758110046,
-          0.839260458946228,
-          0.4206574559211731,
-          0.35542207956314087,
-          0.20313608646392822,
-          0.21616561710834503,
-          0.5085342526435852,
-          0.1830843836069107,
-          0.393907368183136,
-          0.3901851773262024,
-          0.34328851103782654,
-          0.8214281797409058,
-          0.5577329397201538,
-          0.6427444219589233,
-          0.37178701162338257,
-          0.3994154930114746,
-          0.3580907881259918,
-          0.8406038284301758,
-          0.4289478659629822,
-          0.9354329109191895,
-          0.2610803246498108,
-          0.12179839611053467,
-          0.2026183158159256,
-          0.5441363453865051,
-          0.42418813705444336,
-          0.35796743631362915,
-          0.21454016864299774,
-          0.6362805366516113,
-          0.4094250202178955,
-          0.3864958882331848,
-          0.32800084352493286,
-          0.15223217010498047,
-          0.48753517866134644,
-          0.5019508600234985,
-          0.845576286315918,
-          0.7815681099891663,
-          0.5233713388442993,
-          0.34707021713256836,
-          0.26516351103782654,
-          0.5447980761528015,
-          0.569729208946228,
-          0.5469716787338257,
-          0.30339792370796204,
-          0.5697838068008423,
-          0.2694746255874634,
-          0.2848798632621765,
-          0.45616990327835083,
-          0.3615877032279968,
-          0.6941409707069397,
-          0.5672474503517151,
-          0.5404132604598999,
-          0.7676708102226257,
-          0.2914889454841614,
-          0.8202422857284546,
-          0.5488724112510681,
-          0.6117497682571411,
-          0.2768411934375763,
-          0.2580873966217041,
-          0.2866087257862091,
-          0.25340700149536133,
-          0.39905619621276855,
-          0.5676197409629822,
-          0.2999364733695984,
-          0.7428163290023804,
-          0.4202784597873688,
-          0.5675026774406433,
-          0.42694100737571716,
-          0.4559133052825928,
-          0.9446031451225281,
-          0.5738562941551208,
-          0.5707443952560425,
-          0.5453180074691772,
-          0.3223907947540283,
-          0.5991066694259644,
-          0.4520091116428375,
-          0.5416899919509888,
-          0.5794038772583008,
-          0.41604456305503845,
-          0.3481016457080841,
-          0.6949661374092102,
-          0.31202682852745056,
-          0.3425890803337097,
-          0.37019386887550354,
-          0.3918279707431793,
-          0.6375806331634521,
-          1.0107237100601196,
-          0.7948455214500427,
-          0.3794461488723755,
-          0.9569169282913208,
-          0.38022908568382263,
-          0.39158254861831665,
-          0.28092873096466064,
-          0.49257564544677734,
-          0.26403412222862244,
-          0.581787645816803,
-          0.26089993119239807,
-          0.46818771958351135,
-          0.6649788618087769,
-          0.40894460678100586,
-          0.763916015625,
-          0.38571032881736755,
-          0.5845966935157776,
-          0.6800218820571899,
-          0.47281575202941895,
-          0.4429876208305359,
-          0.2453831136226654,
-          0.5786029100418091,
-          0.3747848868370056,
-          0.5507960915565491,
-          0.1960677206516266,
-          0.3365151882171631,
-          0.30705875158309937,
-          0.4584624767303467,
-          0.5215798020362854,
-          0.2194550484418869,
-          0.2778930068016052,
-          0.23652967810630798,
-          0.27437296509742737,
-          0.12692813575267792,
-          0.2907795011997223,
-          0.3903278410434723,
-          0.27809828519821167,
-          0.8016510009765625,
-          0.29485318064689636,
-          0.442976176738739,
-          0.28882941603660583,
-          0.3782258629798889,
-          0.26739203929901123,
-          0.8322014808654785,
-          0.22333034873008728,
-          0.7225127816200256,
-          0.5031032562255859,
-          0.14603176712989807,
-          0.4107421338558197,
-          0.19090536236763,
-          0.2317584753036499,
-          0.21820595860481262,
-          0.6664113998413086,
-          0.34699830412864685,
-          0.5861060619354248,
-          0.5221594572067261,
-          0.37487760186195374,
-          0.20831403136253357,
-          0.42567339539527893,
-          0.15004009008407593,
-          0.547377347946167,
-          0.11534025520086288,
-          0.3091840147972107,
-          0.21701054275035858,
-          0.24749641120433807,
-          0.38849225640296936,
-          0.6236919164657593,
-          0.7785841226577759,
-          0.9336296319961548,
-          0.2950669527053833,
-          0.25050458312034607,
-          0.34149619936943054,
-          0.2238757163286209,
-          0.21536731719970703,
-          0.470775842666626,
-          0.3492281436920166,
-          1.0796414613723755,
-          0.16379037499427795,
-          0.10799019038677216,
-          0.39913442730903625,
-          0.16514906287193298,
-          0.2961501479148865,
-          0.6332513689994812,
-          0.3483124375343323,
-          0.3231945037841797,
-          0.5361443758010864,
-          0.4113987386226654,
-          0.695641279220581,
-          0.1855519860982895,
-          0.19931229948997498,
-          0.27904433012008667,
-          0.4652504324913025,
-          0.4767753481864929,
-          0.3677249550819397,
-          0.26333317160606384,
-          0.36600393056869507,
-          0.42545947432518005,
-          0.5853521227836609,
-          0.2670530676841736,
-          0.14124718308448792,
-          0.2591845393180847,
-          0.10384084284305573,
-          0.08028300106525421,
-          0.7719931602478027,
-          0.5125044584274292,
-          1.124315857887268,
-          0.29444628953933716,
-          0.645147979259491,
-          0.2321443110704422,
-          0.5962263941764832,
-          0.48332419991493225,
-          1.0730419158935547,
-          0.3957912027835846,
-          0.3802769184112549,
-          0.577154815196991,
-          0.4631439745426178,
-          0.969294548034668,
-          0.3340124189853668,
-          0.27889829874038696,
-          0.3518500328063965,
-          0.3195561170578003,
-          0.29832786321640015,
-          0.2729322016239166,
-          0.9382749795913696,
-          0.18967559933662415,
-          0.3467618227005005,
-          0.4656370282173157,
-          0.46193045377731323,
-          0.19111327826976776,
-          0.4193413257598877,
-          0.1126008853316307,
-          0.6352649331092834,
-          0.5832198858261108,
-          0.16809135675430298,
-          0.5647442936897278,
-          0.27541202306747437,
-          0.8396093249320984,
-          0.6124662756919861,
-          0.5793838500976562,
-          0.3841623365879059,
-          0.31533336639404297,
-          0.1467626541852951,
-          0.6446176171302795,
-          0.42055290937423706,
-          0.1736195832490921,
-          0.35788315534591675,
-          0.4834797978401184,
-          1.0282800197601318,
-          0.13835570216178894,
-          0.4655580520629883,
-          0.45540761947631836,
-          0.3080369234085083,
-          0.38043317198753357,
-          0.0694122463464737,
-          0.15990029275417328,
-          0.5399829745292664,
-          0.42322975397109985,
-          0.40007057785987854,
-          0.24551177024841309,
-          0.42177921533584595,
-          0.3354904055595398,
-          0.305682510137558,
-          0.16076476871967316,
-          0.5285727381706238,
-          0.29189127683639526,
-          0.3459763526916504,
-          0.4853358864784241,
-          0.6491932272911072,
-          0.4923343062400818,
-          0.2859308421611786,
-          0.4599456787109375,
-          0.41412320733070374,
-          0.43881258368492126,
-          0.45090270042419434,
-          0.45566844940185547,
-          1.0677928924560547,
-          0.17156289517879486,
-          0.32191553711891174,
-          0.5668681859970093,
-          0.23412999510765076,
-          0.2287621796131134,
-          0.2451322078704834,
-          0.3797767758369446,
-          0.7723601460456848,
-          0.15471571683883667,
-          0.33231115341186523,
-          0.10060479491949081,
-          0.31858882308006287,
-          0.5534226298332214,
-          0.4906136393547058,
-          0.2170345038175583,
-          0.3945397734642029,
-          0.4069906771183014,
-          0.3699072301387787,
-          0.5765577554702759,
-          0.20703521370887756,
-          0.1556961089372635,
-          0.3638346791267395,
-          0.2204539030790329,
-          0.3563755452632904,
-          0.13899393379688263,
-          0.4058627486228943,
-          0.2210249900817871,
-          0.3989500403404236,
-          0.35524681210517883,
-          0.18748818337917328,
-          0.3203352689743042,
-          0.1610220968723297,
-          0.2857784628868103,
-          0.6779120564460754,
-          0.19381950795650482,
-          0.41409629583358765,
-          0.8025749921798706,
-          0.7554987668991089,
-          0.1456572562456131,
-          0.15676842629909515,
-          0.39275380969047546,
-          0.536660373210907,
-          0.20366306602954865,
-          0.3261570930480957,
-          0.2582018971443176,
-          0.21457576751708984,
-          0.6167683005332947,
-          0.439426988363266,
-          0.17834599316120148,
-          0.48090916872024536,
-          1.4856115579605103,
-          0.0915829986333847,
-          0.34598439931869507,
-          0.3562201261520386,
-          0.22693213820457458,
-          0.2718190550804138,
-          0.18724241852760315,
-          0.24289259314537048,
-          0.8717570304870605,
-          0.22201521694660187,
-          0.6017200946807861,
-          0.09056297689676285,
-          0.15231218934059143,
-          0.14399845898151398,
-          0.11619337648153305,
-          0.15978939831256866,
-          0.43446773290634155,
-          0.12164948880672455,
-          0.6501521468162537,
-          0.3507586419582367,
-          0.20334863662719727,
-          0.33664852380752563,
-          0.2729550004005432,
-          0.31600290536880493,
-          0.11929695308208466,
-          0.2001006305217743,
-          0.24858896434307098,
-          0.4709855020046234,
-          0.12933239340782166,
-          0.7611289024353027,
-          0.4049680829048157,
-          0.09919247031211853,
-          0.43667072057724,
-          0.3347734808921814,
-          0.3500886559486389,
-          0.5526772141456604,
-          0.18270298838615417,
-          0.18653002381324768,
-          0.36870893836021423,
-          0.16716769337654114,
-          0.37023672461509705,
-          0.34670960903167725,
-          0.41262149810791016,
-          0.5629145503044128,
-          0.13403025269508362,
-          0.3638155460357666,
-          0.47772112488746643,
-          0.41156020760536194,
-          0.12012441456317902,
-          0.2193690985441208,
-          0.20217227935791016,
-          0.09769073128700256,
-          0.7282608151435852,
-          0.10522177070379257,
-          0.9661720991134644,
-          0.06811663508415222,
-          0.2336260974407196,
-          0.15692603588104248,
-          0.641910970211029,
-          0.5042229890823364,
-          0.19089405238628387,
-          0.3258182108402252,
-          0.151984304189682,
-          0.3439784049987793,
-          0.3531765341758728,
-          0.4698052406311035,
-          0.5141698718070984,
-          0.36121004819869995,
-          0.32519805431365967,
-          0.16760312020778656,
-          0.329439252614975,
-          0.33139458298683167,
-          0.5531335473060608,
-          0.4874722361564636,
-          0.2709265947341919,
-          0.3243344724178314,
-          0.4630916118621826,
-          0.15138842165470123,
-          0.3524761497974396,
-          0.26848000288009644,
-          0.4023500978946686,
-          0.3167988955974579,
-          0.12778529524803162
+          95.13279485702515,
+          83.86075180768967,
+          72.27339777350426,
+          65.92202863097191,
+          63.59204542636871,
+          59.00900012254715,
+          58.88214838504791,
+          56.96866858005524,
+          54.19743275642395,
+          53.16101652383804
         ],
         "device": {
           "_type": "unknown",
           "repr": "cuda:0"
         },
         "verbose": false,
-        "model_name": "AttrMaskMolecularEncoder"
+        "model_name": "MoamaMolecularEncoder"
       },
       "metrics": {}
     }
   },
-  "last_updated": "2025-07-31T11:23:16.269033"
+  "last_updated": "2025-08-07T01:00:46.754276"
 }