Upload unet_cifar.yaml with huggingface_hub
Browse files- unet_cifar.yaml +23 -0
unet_cifar.yaml
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# UNet + CFM training hyperparameters (used by train_cfm_unet.py --config)
|
| 2 |
+
|
| 3 |
+
sigma: 0.0
|
| 4 |
+
|
| 5 |
+
# Image shape C, H, W
|
| 6 |
+
dim: [3, 32, 32]
|
| 7 |
+
|
| 8 |
+
lr: 1.0e-4
|
| 9 |
+
weight_decay: 0.0
|
| 10 |
+
|
| 11 |
+
# NeuralODE visualization / sampling
|
| 12 |
+
save_ep: 30
|
| 13 |
+
inference_steps: 100
|
| 14 |
+
vis_batch_size: 4
|
| 15 |
+
|
| 16 |
+
# UNet (torchcfm UNetModelWrapper)
|
| 17 |
+
num_res_blocks: 2
|
| 18 |
+
num_channels: 128
|
| 19 |
+
channel_mult: [1, 2, 2, 2]
|
| 20 |
+
num_heads: 4
|
| 21 |
+
num_head_channels: 64
|
| 22 |
+
attention_resolutions: "16"
|
| 23 |
+
dropout: 0.1
|