| 102 | |
| 103 | @dataclass |
| 104 | class State: |
| 105 | generator: DAC |
| 106 | optimizer_g: AdamW |
| 107 | scheduler_g: ExponentialLR |
| 108 | |
| 109 | stft_loss: losses.MultiScaleSTFTLoss |
| 110 | mel_loss: losses.MelSpectrogramLoss |
| 111 | waveform_loss: losses.L1Loss |
| 112 | |
| 113 | train_data: AudioDataset |
| 114 | val_data: AudioDataset |
| 115 | |
| 116 | tracker: Tracker |
| 117 | |
| 118 | |
| 119 | @argbind.bind(without_prefix=True) |