miittnnss
/

testmodel

Model card Files Files and versions

miittnnss commited on Oct 15, 2023

Commit

1e65589

·

1 Parent(s): 71e042e

Update config.json

Files changed (1) hide show

config.json +3 -1

config.json CHANGED Viewed

@@ -27,7 +27,9 @@
         // Add discriminator configuration here if needed
     },
     "optimizer": {
-        // Add optimizer configuration here if needed
     },
     "loss": "BCELoss"
 }

         // Add discriminator configuration here if needed
     },
     "optimizer": {
+        "type": "Adam",
+        "learning_rate": 0.0002,
+        "betas": [0.5, 0.999]
     },
     "loss": "BCELoss"
 }