ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 24 days ago

Commit

87aa12d

·

verified ·

1 Parent(s): 6b55e58

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -85,7 +85,7 @@
     "dim_feedforward": 2048,
     "diffusion_step_embed_dim": 128,
     "num_inference_steps": 20,
-    "optimizer_lr": 0.0001,
     "optimizer_betas": [
         0.95,
         0.999

     "dim_feedforward": 2048,
     "diffusion_step_embed_dim": 128,
     "num_inference_steps": 20,
+    "optimizer_lr": 0.000014,
     "optimizer_betas": [
         0.95,
         0.999