TimeRobber commited on
Commit
0f005f8
1 Parent(s): 0904434

Update vae/config.json

Browse files

Update the vae sample size to reflect the actual size that one should use.

Files changed (1) hide show
  1. vae/config.json +1 -1
vae/config.json CHANGED
@@ -20,7 +20,7 @@
20
  "layers_per_block": 2,
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
- "sample_size": 256,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
26
  "UpDecoderBlock2D",
 
20
  "layers_per_block": 2,
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
+ "sample_size": 512,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
26
  "UpDecoderBlock2D",