upload vomp

ft/geometry_transformer/ckpts/geometry_encoder_ema0.9999_step0200000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e726a661abd354756383ded8a68624f6d984f344463f1d67816e71959223107
+size 343249527

ft/geometry_transformer/ckpts/geometry_encoder_step0200000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b5ff2498751d4a431dae4486312a294ba2723ea332e5eaf01bd461f87f746c1
+size 343249477

ft/geometry_transformer/ckpts/misc_step0200000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c74ef3ddd25c5cf037ace1e378638320b6182bbfcb5ae3d84c02f0b44e997306
+size 686471123

ft/geometry_transformer/evaluation_results.json

@@ -0,0 +1,557 @@
+{
+    "val_loss": 0.13122641187876724,
+    "val_loss_youngs": 0.0627478299173239,
+    "val_loss_poisson": 0.05093700650383816,
+    "val_loss_density": 0.017541575696781667,
+    "val_samples": 166,
+    "val_rel_error_log_youngs": 0.04208697751164436,
+    "val_rel_error_log_youngs_std": 0.05111011490225792,
+    "val_rel_error_poisson": 0.08366880565881729,
+    "val_rel_error_poisson_std": 0.034644145518541336,
+    "val_rel_error_density": 0.09081185609102249,
+    "val_rel_error_density_std": 0.1383020132780075,
+    "val_rel_error_log_density": 0.013480736874043941,
+    "val_rel_error_log_density_std": 0.015981599688529968,
+    "val_abs_error_youngs": 34292217856.0,
+    "val_abs_error_youngs_std": 66275778560.0,
+    "val_abs_error_log_youngs": 0.37646999955177307,
+    "val_abs_error_log_youngs_std": 0.39179643988609314,
+    "val_abs_error_poisson": 0.024959132075309753,
+    "val_abs_error_poisson_std": 0.00951018650084734,
+    "val_abs_error_density": 113.30818176269531,
+    "val_abs_error_density_std": 301.20452880859375,
+    "val_rel_error_overall": 0.04641217365860939,
+    "val_rel_error_overall_std": 0.04675865173339844,
+    "per_object_rel_error_log_youngs_mean": 0.04090757668018341,
+    "per_object_rel_error_log_youngs_std": 0.036562226712703705,
+    "per_object_rel_error_log_youngs_min": 0.0006934626726433635,
+    "per_object_rel_error_log_youngs_max": 0.18356622755527496,
+    "per_object_rel_error_poisson_mean": 0.08175762742757797,
+    "per_object_rel_error_poisson_std": 0.028551137074828148,
+    "per_object_rel_error_poisson_min": 0.017800521105527878,
+    "per_object_rel_error_poisson_max": 0.1892934888601303,
+    "per_object_rel_error_density_mean": 0.09207938611507416,
+    "per_object_rel_error_density_std": 0.06719443202018738,
+    "per_object_rel_error_density_min": 0.0037892332766205072,
+    "per_object_rel_error_density_max": 0.7864506840705872,
+    "per_object_rel_error_log_density_mean": 0.01332605630159378,
+    "per_object_rel_error_log_density_std": 0.008755325339734554,
+    "per_object_rel_error_log_density_min": 0.00047562114195898175,
+    "per_object_rel_error_log_density_max": 0.0910286232829094,
+    "per_object_rel_error_overall_mean": 0.04533041641116142,
+    "per_object_rel_error_overall_std": 0.017195673659443855,
+    "per_object_rel_error_overall_min": 0.017565801739692688,
+    "per_object_rel_error_overall_max": 0.1020798459649086,
+    "per_object_abs_error_log_youngs_mean": 0.37931695580482483,
+    "per_object_abs_error_log_youngs_std": 0.2906107008457184,
+    "per_object_abs_error_poisson_mean": 0.02405145950615406,
+    "per_object_abs_error_poisson_std": 0.007175541948527098,
+    "per_object_abs_error_density_mean": 142.60604858398438,
+    "per_object_abs_error_density_std": 166.91015625,
+    "per_object_displacement_errors": {
+        "youngs": [
+            0.06349022686481476,
+            0.43226727843284607,
+            0.055241264402866364,
+            0.048667341470718384,
+            0.2813689410686493,
+            0.060806743800640106,
+            0.361782044172287,
+            0.03818412125110626,
+            0.5428236722946167,
+            0.06199537590146065,
+            0.04310586676001549,
+            0.02182183787226677,
+            0.012189667671918869,
+            0.0906536877155304,
+            0.5241801142692566,
+            0.059930771589279175,
+            0.046151675283908844,
+            0.08723264932632446,
+            0.1892584264278412,
+            0.040759265422821045,
+            0.05124332010746002,
+            0.2167881280183792,
+            0.06390843540430069,
+            0.02652408741414547,
+            0.06233308091759682,
+            0.9542932510375977,
+            0.03835977241396904,
+            0.05756330490112305,
+            0.038225796073675156,
+            0.9518799185752869,
+            0.006241165101528168,
+            0.052862927317619324,
+            0.062032606452703476,
+            0.3034905195236206,
+            0.34506168961524963,
+            0.05719046667218208,
+            0.035752393305301666,
+            0.029787136241793633,
+            0.061392154544591904,
+            0.017351441085338593,
+            0.4527038633823395,
+            0.052052102982997894,
+            0.3703787326812744,
+            0.015236212871968746,
+            0.21648423373699188,
+            0.32874035835266113,
+            0.4393019378185272,
+            0.4800335764884949,
+            0.3468877375125885,
+            0.010730811394751072,
+            0.5224068760871887,
+            0.9138173460960388,
+            0.3867100179195404,
+            0.36680445075035095,
+            0.32877105474472046,
+            0.3062923550605774,
+            0.32816430926322937,
+            0.23283866047859192,
+            0.5490462779998779,
+            0.5282837748527527,
+            0.20703627169132233,
+            0.23900127410888672,
+            1.2707161903381348,
+            1.27230966091156,
+            1.269689917564392,
+            1.2876509428024292,
+            0.30226659774780273,
+            0.6465096473693848,
+            0.4620765447616577,
+            0.4434468448162079,
+            0.3033030927181244,
+            0.31035006046295166,
+            0.31787818670272827,
+            0.15458008646965027,
+            0.7471697926521301,
+            0.19751501083374023,
+            0.957854688167572,
+            0.953911542892456,
+            0.37851083278656006,
+            0.420046865940094,
+            0.3659699261188507,
+            0.6739988923072815,
+            0.4484369456768036,
+            0.42170676589012146,
+            0.43818339705467224,
+            0.847148597240448,
+            0.4620090126991272,
+            0.2868402898311615,
+            0.3319235146045685,
+            0.4809403717517853,
+            0.42133402824401855,
+            0.41438037157058716,
+            0.389771431684494,
+            0.5223903656005859,
+            0.40300217270851135,
+            0.35892271995544434,
+            0.9139617681503296,
+            0.7023851275444031,
+            0.691601574420929,
+            0.6898798942565918,
+            0.08320830017328262,
+            0.6866270303726196,
+            0.4227645695209503,
+            1.0561813116073608,
+            0.908888041973114,
+            0.3009556233882904,
+            0.2910342216491699,
+            0.31270337104797363,
+            0.5789288878440857,
+            0.41381457448005676,
+            0.4162984788417816,
+            0.41571861505508423,
+            0.3264140486717224,
+            0.3267183005809784,
+            0.2985910177230835,
+            0.3196257948875427,
+            0.3039432168006897,
+            0.3463340699672699,
+            0.6184525489807129,
+            0.5721779465675354,
+            1.213032841682434,
+            0.38657617568969727,
+            0.3908078372478485,
+            0.33228808641433716,
+            0.3079962134361267,
+            0.31585392355918884,
+            0.19023211300373077,
+            0.3904349207878113,
+            0.3837205767631531,
+            0.529259443283081,
+            1.037075161933899,
+            0.2258569300174713,
+            0.44219374656677246,
+            0.1097864881157875,
+            0.689801812171936,
+            0.7316137552261353,
+            0.8113479018211365,
+            0.42132896184921265,
+            0.4008280038833618,
+            0.3570237159729004,
+            0.4500788152217865,
+            0.33947375416755676,
+            0.33514514565467834,
+            0.40605446696281433,
+            0.3283573389053345,
+            0.404716432094574,
+            0.37342509627342224,
+            1.1277010440826416,
+            0.19048243761062622,
+            0.2358877956867218,
+            0.1047753393650055,
+            0.18585142493247986,
+            0.14472989737987518,
+            0.1625816822052002,
+            0.442467600107193,
+            0.2656879127025604,
+            0.4058883786201477,
+            0.28703397512435913,
+            0.16932347416877747,
+            0.4325987994670868,
+            0.10700111091136932,
+            0.19669124484062195,
+            0.28105300664901733,
+            0.47726452350616455,
+            0.11719958484172821,
+            0.25221338868141174
+        ],
+        "poisson": [
+            0.026263300329446793,
+            0.020084770396351814,
+            0.027153050526976585,
+            0.027348870411515236,
+            0.029721641913056374,
+            0.026478594169020653,
+            0.02289462648332119,
+            0.02121601440012455,
+            0.022701002657413483,
+            0.032623741775751114,
+            0.028136907145380974,
+            0.020158909261226654,
+            0.01575593277812004,
+            0.022123923525214195,
+            0.022000255063176155,
+            0.026431433856487274,
+            0.02822195179760456,
+            0.03021322749555111,
+            0.014689691364765167,
+            0.027630776166915894,
+            0.027173368260264397,
+            0.012853695079684258,
+            0.02634849026799202,
+            0.028105024248361588,
+            0.023846954107284546,
+            0.02220754697918892,
+            0.021534916013479233,
+            0.02658773958683014,
+            0.02758135460317135,
+            0.02396511845290661,
+            0.03463847190141678,
+            0.026653604581952095,
+            0.026312775909900665,
+            0.025295764207839966,
+            0.030033672228455544,
+            0.02685457095503807,
+            0.028592940419912338,
+            0.02949104830622673,
+            0.02612273581326008,
+            0.03572564944624901,
+            0.01967998966574669,
+            0.027013206854462624,
+            0.012286133132874966,
+            0.01960734650492668,
+            0.012228275649249554,
+            0.02112737111747265,
+            0.02026955597102642,
+            0.025230951607227325,
+            0.02122294344007969,
+            0.034729357808828354,
+            0.02570393681526184,
+            0.022578319534659386,
+            0.021828902885317802,
+            0.02614022046327591,
+            0.02599526010453701,
+            0.020643847063183784,
+            0.02119636908173561,
+            0.019625257700681686,
+            0.021973975002765656,
+            0.021579260006546974,
+            0.021341867744922638,
+            0.021174702793359756,
+            0.04000457748770714,
+            0.03977000340819359,
+            0.039606161415576935,
+            0.037267692387104034,
+            0.03642662987112999,
+            0.056867215782403946,
+            0.01908111944794655,
+            0.01988123543560505,
+            0.02549837715923786,
+            0.014428195543587208,
+            0.01429420243948698,
+            0.01644865609705448,
+            0.04086029902100563,
+            0.01334102638065815,
+            0.021664388477802277,
+            0.02230629324913025,
+            0.023060377687215805,
+            0.023976827040314674,
+            0.03029419481754303,
+            0.02242189459502697,
+            0.03027327172458172,
+            0.023802567273378372,
+            0.02051856927573681,
+            0.03039618209004402,
+            0.02681339904665947,
+            0.014543203637003899,
+            0.025259295478463173,
+            0.02512430027127266,
+            0.025190792977809906,
+            0.02543262019753456,
+            0.024359924718737602,
+            0.025705378502607346,
+            0.02380271628499031,
+            0.0238484013825655,
+            0.02254480868577957,
+            0.027035849168896675,
+            0.02846434712409973,
+            0.02851000986993313,
+            0.018721003085374832,
+            0.029302844777703285,
+            0.024115771055221558,
+            0.013261550106108189,
+            0.008262566290795803,
+            0.01673758402466774,
+            0.01740625686943531,
+            0.028064345940947533,
+            0.027420664206147194,
+            0.024725815281271935,
+            0.025791822001338005,
+            0.024104401469230652,
+            0.01954149268567562,
+            0.016637632623314857,
+            0.016809629276394844,
+            0.016713757067918777,
+            0.017723312601447105,
+            0.017125282436609268,
+            0.02366962842643261,
+            0.023332668468356133,
+            0.032102134078741074,
+            0.038022231310606,
+            0.006230182480067015,
+            0.016106436029076576,
+            0.017901711165905,
+            0.02520536072552204,
+            0.01406165026128292,
+            0.026813236996531487,
+            0.0270253736525774,
+            0.0317155160009861,
+            0.0365779735147953,
+            0.017406191676855087,
+            0.03795390576124191,
+            0.047356270253658295,
+            0.02865751087665558,
+            0.02899571694433689,
+            0.029932668432593346,
+            0.024864433333277702,
+            0.025023264810442924,
+            0.021587982773780823,
+            0.03189650923013687,
+            0.016688717529177666,
+            0.014066935516893864,
+            0.025492660701274872,
+            0.016280431300401688,
+            0.01822902262210846,
+            0.020818715915083885,
+            0.03198768198490143,
+            0.011380918323993683,
+            0.019861556589603424,
+            0.02263670414686203,
+            0.027694717049598694,
+            0.023497575893998146,
+            0.016479160636663437,
+            0.012524774298071861,
+            0.02335835061967373,
+            0.011590931564569473,
+            0.021977752447128296,
+            0.01710539683699608,
+            0.012583867646753788,
+            0.023838097229599953,
+            0.023481691256165504,
+            0.024655578657984734,
+            0.023864906281232834,
+            0.023145074024796486,
+            0.02032037079334259
+        ],
+        "density": [
+            80.13394927978516,
+            110.5985336303711,
+            66.58460998535156,
+            65.41122436523438,
+            198.61260986328125,
+            76.99887084960938,
+            199.49261474609375,
+            46.46979522705078,
+            93.89729309082031,
+            22.000429153442383,
+            53.88482666015625,
+            45.76938247680664,
+            29.209144592285156,
+            52.68577194213867,
+            91.20830535888672,
+            77.39519500732422,
+            46.07134246826172,
+            180.36459350585938,
+            95.11202239990234,
+            63.34445571899414,
+            66.76737976074219,
+            262.5780944824219,
+            77.47183990478516,
+            59.78453826904297,
+            81.88551330566406,
+            105.51189422607422,
+            53.1313362121582,
+            76.13691711425781,
+            65.02871704101562,
+            40.15388870239258,
+            38.54841613769531,
+            72.90275573730469,
+            79.71503448486328,
+            228.22659301757812,
+            156.18228149414062,
+            71.69425964355469,
+            50.80032730102539,
+            35.447265625,
+            83.28423309326172,
+            51.92814636230469,
+            149.42727661132812,
+            70.55064392089844,
+            270.56793212890625,
+            30.202396392822266,
+            271.6907043457031,
+            202.6064453125,
+            118.79430389404297,
+            11.280104637145996,
+            207.51004028320312,
+            39.73664093017578,
+            13.974124908447266,
+            113.9129409790039,
+            426.60858154296875,
+            416.8294677734375,
+            201.83705139160156,
+            183.47613525390625,
+            186.5145721435547,
+            102.67427062988281,
+            85.05398559570312,
+            86.49317932128906,
+            170.59136962890625,
+            174.4372100830078,
+            85.45082092285156,
+            126.8759994506836,
+            92.36326599121094,
+            94.3072738647461,
+            93.27640533447266,
+            110.65652465820312,
+            124.4757308959961,
+            124.88130187988281,
+            173.42758178710938,
+            244.44830322265625,
+            210.07920837402344,
+            114.15705108642578,
+            224.93136596679688,
+            649.417236328125,
+            161.07395935058594,
+            106.2195816040039,
+            228.7470703125,
+            156.16163635253906,
+            104.87094116210938,
+            38.49231719970703,
+            268.54339599609375,
+            154.3653564453125,
+            73.601806640625,
+            135.85589599609375,
+            203.38868713378906,
+            231.6894073486328,
+            264.414794921875,
+            10.79931640625,
+            175.9723358154297,
+            206.63623046875,
+            157.7445831298828,
+            14.298713684082031,
+            167.0341033935547,
+            69.62113952636719,
+            113.1507797241211,
+            216.60440063476562,
+            227.80023193359375,
+            240.06350708007812,
+            406.7103576660156,
+            225.3424530029297,
+            123.29544067382812,
+            133.20179748535156,
+            149.52792358398438,
+            9.085606575012207,
+            24.496944427490234,
+            82.90860748291016,
+            112.8775405883789,
+            161.7285614013672,
+            313.5748291015625,
+            87.8735122680664,
+            120.53755187988281,
+            75.94342803955078,
+            12.994876861572266,
+            119.22789001464844,
+            97.30532836914062,
+            45.280521392822266,
+            122.3984375,
+            271.1710205078125,
+            162.40538024902344,
+            417.96002197265625,
+            70.09521484375,
+            99.54105377197266,
+            270.3040466308594,
+            221.57627868652344,
+            170.87918090820312,
+            143.16505432128906,
+            132.18809509277344,
+            798.4971313476562,
+            188.36207580566406,
+            94.0706558227539,
+            271.4931640625,
+            1771.27587890625,
+            232.3736114501953,
+            84.54676818847656,
+            264.2999267578125,
+            124.2440414428711,
+            174.03526306152344,
+            195.28773498535156,
+            276.59783935546875,
+            101.7153091430664,
+            40.5678596496582,
+            176.60690307617188,
+            55.76715850830078,
+            51.19596862792969,
+            14.554147720336914,
+            172.27920532226562,
+            64.28573608398438,
+            63.719905853271484,
+            48.371891021728516,
+            133.43431091308594,
+            40.33542251586914,
+            51.7469482421875,
+            43.46722412109375,
+            64.22266387939453,
+            51.130943298339844,
+            62.42725372314453,
+            40.448570251464844,
+            45.70907974243164,
+            48.77022171020508,
+            48.36052322387695,
+            48.0938606262207,
+            53.0426025390625,
+            48.39015579223633,
+            62.1863899230957
+        ]
+    }
+}

ft/matvae2/checkpoints/checkpoint_851/custom_checkpoint_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f622e3c5a07f4ce714d0de5722b367082314db533edffc2f3baa39777e4305aa
+size 1240

ft/matvae2/checkpoints/checkpoint_851/custom_checkpoint_1.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:625bea504e46688a1ae7c0ceb4065b53420169b636d9bc67991a5f1930e00475
+size 1240

ft/matvae2/checkpoints/checkpoint_851/custom_checkpoint_2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4203470b746c23fc434fceefb67ef86c8aa4896dabb5696c80e5146bc479acc
+size 912

ft/matvae2/checkpoints/checkpoint_851/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
+size 1620768

ft/matvae2/checkpoints/checkpoint_851/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f209ae0cebcff7282c24ff3a9f58c6e334e20742de081c0e742c0e29ce8d388c
+size 3283962

ft/matvae2/checkpoints/checkpoint_851/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0a4d3207de77b01743424e557783061b3488dcf3221ab49ebbe8ac81999e710
+size 14408

ft/matvae2/checkpoints/checkpoint_852/custom_checkpoint_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f622e3c5a07f4ce714d0de5722b367082314db533edffc2f3baa39777e4305aa
+size 1240

ft/matvae2/checkpoints/checkpoint_852/custom_checkpoint_1.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:625bea504e46688a1ae7c0ceb4065b53420169b636d9bc67991a5f1930e00475
+size 1240

ft/matvae2/checkpoints/checkpoint_852/custom_checkpoint_2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41f069fc75df34dc2eee7887119a77648b277409f06bf218ab675eb33d0c57c3
+size 912

ft/matvae2/checkpoints/checkpoint_852/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
+size 1620768

ft/matvae2/checkpoints/checkpoint_852/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f209ae0cebcff7282c24ff3a9f58c6e334e20742de081c0e742c0e29ce8d388c
+size 3283962

ft/matvae2/checkpoints/checkpoint_852/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0a4d3207de77b01743424e557783061b3488dcf3221ab49ebbe8ac81999e710
+size 14408

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f622e3c5a07f4ce714d0de5722b367082314db533edffc2f3baa39777e4305aa
+size 1240

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_1.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:625bea504e46688a1ae7c0ceb4065b53420169b636d9bc67991a5f1930e00475
+size 1240

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41f069fc75df34dc2eee7887119a77648b277409f06bf218ab675eb33d0c57c3
+size 912

ft/matvae2/checkpoints/checkpoint_853/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
+size 1620768

ft/matvae2/checkpoints/checkpoint_853/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f209ae0cebcff7282c24ff3a9f58c6e334e20742de081c0e742c0e29ce8d388c
+size 3283962

ft/matvae2/checkpoints/checkpoint_853/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0a4d3207de77b01743424e557783061b3488dcf3221ab49ebbe8ac81999e710
+size 14408

ft/matvae2/final_model/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
+size 1620768

ft/matvae2/normalization_params.json

@@ -0,0 +1,12 @@
+{
+  "E_min": 1000000.0,
+  "E_max": 999378321408.0,
+  "nu_min": 0.0,
+  "nu_max": 0.49000000953674316,
+  "rho_min": 102.56499481201172,
+  "rho_max": 19800.0,
+  "normalization_type": "log_minmax",
+  "dataset_size": 101517,
+  "train_size": 91365,
+  "val_size": 10152
+}

geometry_transformer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b5ff2498751d4a431dae4486312a294ba2723ea332e5eaf01bd461f87f746c1
+size 343249477

inference.json

@@ -0,0 +1,86 @@
+{
+    "geometry_checkpoint_dir": "weights/geometry_transformer.pt",
+    "matvae_checkpoint_dir": "weights/matvae.safetensors",
+    "normalization_params_path": "weights/normalization_params.json",
+    "matvae_checkpoint": "weights/matvae.safetensors",
+    "trellis_weights_path": "weights/TRELLIS-image-large",
+    "models": {
+        "geometry_encoder": {
+            "name": "geometry_encoder",
+            "args": {
+                "resolution": 64,
+                "in_channels": 1024,
+                "model_channels": 768,
+                "latent_channels": 2,
+                "num_blocks": 12,
+                "num_heads": 12,
+                "mlp_ratio": 4,
+                "attn_mode": "swin",
+                "window_size": 8,
+                "use_fp16": true
+            }
+        },
+        "matvae": {
+            "name": "matvae",
+            "args": {
+                "width": 256,
+                "depth": 3,
+                "z_dim": 2,
+                "p_drop": 0.05,
+                "use_learned_variances": false,
+                "use_additional_losses": true
+            }
+        }
+    },
+    "dataset": {
+        "name": "SparseVoxelMaterials",
+        "normalization_type": "log_minmax",
+        "args": {
+            "roots": "datasets/simready",
+            "image_size": 512,
+            "model": "dinov2_vitl14_reg",
+            "resolution": 64,
+            "min_aesthetic_score": 0.0,
+            "max_num_voxels": 32768,
+            "compute_material_stats": false
+        }
+    },
+    "trainer": {
+        "name": "SLatVaeMaterialsTrainer",
+        "args": {
+            "max_steps": 1000000,
+            "batch_size_per_gpu": 16,
+            "batch_split": 1,
+            "optimizer": {
+                "name": "AdamW",
+                "args": {
+                    "lr": 1e-4,
+                    "weight_decay": 0.0
+                }
+            },
+            "ema_rate": [
+                0.9999
+            ],
+            "fp16_mode": "inflat_all",
+            "fp16_scale_growth": 0.001,
+            "elastic": {
+                "name": "LinearMemoryController",
+                "args": {
+                    "target_ratio": 0.75,
+                    "max_mem_ratio_start": 0.5
+                }
+            },
+            "grad_clip": {
+                "name": "AdaptiveGradClipper",
+                "args": {
+                    "max_norm": 1.0,
+                    "clip_percentile": 95
+                }
+            },
+            "i_log": 10,
+            "i_save": 2000,
+            "i_eval": 1000,
+            "loss_type": "l1"
+        }
+    }
+}

matvae.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
+size 1620768

normalization_params.json

@@ -0,0 +1,12 @@
+{
+  "E_min": 1000000.0,
+  "E_max": 999378321408.0,
+  "nu_min": 0.0,
+  "nu_max": 0.49000000953674316,
+  "rho_min": 102.56499481201172,
+  "rho_max": 19800.0,
+  "normalization_type": "log_minmax",
+  "dataset_size": 101517,
+  "train_size": 91365,
+  "val_size": 10152
+}