Spaces:
Running
Running
update
Browse files
examples/nx_clean_unet/yaml/config.yaml
CHANGED
|
@@ -12,7 +12,7 @@ down_sampling_hidden_channels: 64
|
|
| 12 |
down_sampling_kernel_size: 4
|
| 13 |
down_sampling_stride: 2
|
| 14 |
|
| 15 |
-
tsfm_hidden_size:
|
| 16 |
tsfm_attention_heads: 8
|
| 17 |
tsfm_num_blocks: 6
|
| 18 |
tsfm_dropout_rate: 0.1
|
|
|
|
| 12 |
down_sampling_kernel_size: 4
|
| 13 |
down_sampling_stride: 2
|
| 14 |
|
| 15 |
+
tsfm_hidden_size: 256
|
| 16 |
tsfm_attention_heads: 8
|
| 17 |
tsfm_num_blocks: 6
|
| 18 |
tsfm_dropout_rate: 0.1
|