Veldrovive commited on
Commit
55a012d
1 Parent(s): 3066539

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -4
config.json CHANGED
@@ -19,7 +19,7 @@
19
  "channels": 3,
20
  "timesteps": 1000,
21
  "loss_type": "l2",
22
- "beta_schedule": "cosine",
23
  "learned_variance": true
24
  },
25
  "data": {
@@ -49,7 +49,7 @@
49
  },
50
  "train": {
51
  "epochs": 40,
52
- "lr":1e-4,
53
  "wd": 0.01,
54
  "max_grad_norm": 0.5,
55
  "save_every_n_samples": 200000,
@@ -59,6 +59,7 @@
59
  "validation_samples": 100000,
60
  "use_ema": true,
61
  "ema_beta": 0.99,
 
62
  "save_all": false,
63
  "save_latest": true,
64
  "save_best": true,
@@ -86,8 +87,8 @@
86
  "source": "wandb",
87
 
88
  "run_path": "nousr_laion/dalle2_train_decoder/3rrsm5gy",
89
- "file_path": "latest.pth",
90
 
91
- "resume": true
92
  }
93
  }
 
19
  "channels": 3,
20
  "timesteps": 1000,
21
  "loss_type": "l2",
22
+ "beta_schedule": ["cosine"],
23
  "learned_variance": true
24
  },
25
  "data": {
 
49
  },
50
  "train": {
51
  "epochs": 40,
52
+ "lr":5e-5,
53
  "wd": 0.01,
54
  "max_grad_norm": 0.5,
55
  "save_every_n_samples": 200000,
 
59
  "validation_samples": 100000,
60
  "use_ema": true,
61
  "ema_beta": 0.99,
62
+ "ema_param_or_buffer_names_no_ema": ["_dummy"],
63
  "save_all": false,
64
  "save_latest": true,
65
  "save_best": true,
 
87
  "source": "wandb",
88
 
89
  "run_path": "nousr_laion/dalle2_train_decoder/3rrsm5gy",
90
+ "file_path": "best.pth",
91
 
92
+ "resume": false
93
  }
94
  }