Spaces:

qgyd2021
/

nx_denoise

Building

HoneyTian commited on Mar 3

Commit

6de113d

1 Parent(s): 9a47ac9

update

Files changed (1) hide show

examples/nx_clean_unet/yaml/config.yaml CHANGED Viewed

@@ -12,15 +12,15 @@ down_sampling_hidden_channels: 64
 down_sampling_kernel_size: 4
 down_sampling_stride: 2
-tsfm_hidden_size: 1024
-tsfm_attention_heads: 8
 tsfm_num_blocks: 6
 tsfm_dropout_rate: 0.1
 tsfm_max_length: 1024
 tsfm_chunk_size: 1
 tsfm_num_left_chunks: 128
-discriminator_dim: 16
 discriminator_in_channel: 2
 compress_factor: 0.3

 down_sampling_kernel_size: 4
 down_sampling_stride: 2
+tsfm_hidden_size: 256
+tsfm_attention_heads: 4
 tsfm_num_blocks: 6
 tsfm_dropout_rate: 0.1
 tsfm_max_length: 1024
 tsfm_chunk_size: 1
 tsfm_num_left_chunks: 128
+discriminator_dim: 32
 discriminator_in_channel: 2
 compress_factor: 0.3