8ed9309 b27ed9f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
model_name: "clean_unet" channels_input: 1 channels_output: 1 channels_h: 64 max_h: 768 encoder_n_layers: 8 kernel_size: 4 stride: 2 tsfm_n_layers: 5 tsfm_n_head: 8 tsfm_d_model: 512 tsfm_d_inner: 2048
