ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 2 days ago

Commit

9364d1b

·

verified ·

1 Parent(s): 0c77efe

Update config.json

Files changed (1) hide show

config.json +1 -3

config.json CHANGED Viewed

@@ -56,7 +56,6 @@
     "nhead": 4,
     "num_encoder_layers": 4,
     "dim_feedforward": 512,
-    "state_token_dim": 64,
     "optimizer_lr": 0.0001,
     "optimizer_betas": [
         0.95,
@@ -65,6 +64,5 @@
     "optimizer_eps": 1e-08,
     "optimizer_weight_decay": 1e-06,
     "scheduler_name": "cosine",
-    "scheduler_warmup_steps": 500,
-    "do_mask_loss_for_padding": true
 }

     "nhead": 4,
     "num_encoder_layers": 4,
     "dim_feedforward": 512,
     "optimizer_lr": 0.0001,
     "optimizer_betas": [
         0.95,
     "optimizer_eps": 1e-08,
     "optimizer_weight_decay": 1e-06,
     "scheduler_name": "cosine",
+    "scheduler_warmup_steps": 500
 }