| model: | |
| in_channels: 2 | |
| out_channels: 2 | |
| dims: | |
| - 96 | |
| - 192 | |
| - 384 | |
| - 768 | |
| depths: | |
| - 2 | |
| - 2 | |
| - 4 | |
| - 2 | |
| drop_path: 0 | |
| time_dim: 256 | |
| cond_dim: 384 | |
| total_freq_bins: 512 | |
| hr_freq_bins: 432 | |
| feature_enc_layers: 4 | |
| cond_dropout_prob: 0.1 | |
| sr_to_lr_bins: | |
| 8: 80 | |
| 12: 128 | |
| 16: 170 | |
| 24: 256 | |
| transform: | |
| window_fn: hann | |
| n_fft: 1024 | |
| sampling_rate: 48000 | |
| hop_length: 512 | |
| alpha: 0.2 | |
| beta: 1 | |
| comp_eps: 0.0001 | |
| path: | |
| class_path: universr.flow.path.OriginalCFMPath | |
| init_args: | |
| sigma_min: 0.0001 | |