| { | |
| "etas_end": 0.99, | |
| "flow_kwargs": { | |
| "pretrained_path": null | |
| }, | |
| "flow_model": "raft", | |
| "kappa": 2.0, | |
| "min_noise_level": 0.04, | |
| "p": 0.3, | |
| "synthesis_kwargs": { | |
| "channels": [ | |
| 128, | |
| 256, | |
| 384, | |
| 512 | |
| ], | |
| "dropout": 0.0, | |
| "expansion_rate": 1.5, | |
| "grid_attn": true, | |
| "heads": 1, | |
| "in_channels": 3, | |
| "num_conv_blocks": 1, | |
| "temb_channels": 512, | |
| "window_attn": true, | |
| "window_size": 8 | |
| }, | |
| "timesteps": 20, | |
| "warping_kwargs": { | |
| "channels": [ | |
| 128, | |
| 256, | |
| 384, | |
| 512 | |
| ], | |
| "in_channels": 3 | |
| } | |
| } |