(self, samples)
| 95 | return sample |
| 96 | |
| 97 | def collater(self, samples): |
| 98 | batch = super(ShallowDiffusionOfflineDataset, self).collater(samples) |
| 99 | if self.prefix != 'train' and hparams['fs2_ckpt'] != '': |
| 100 | batch['fs2_mels'] = utils.collate_2d([s['fs2_mel'] for s in samples], 0.0) |
| 101 | return batch |
| 102 | |
| 103 | |
| 104 | class DiffSingerOfflineTask(DiffSingerTask): |
nothing calls this directly
no outgoing calls
no test coverage detected