Amphion
Amphion copied to clipboard
[Help]: Natural Speech 2 training issue(Data loader)
During training, I printed the data output from the dataset, NS2Collator, and DataLoader separately. Only the DataLoader output data showed anomalies (some tensors were all zeros). The duration data is loaded through the dataset, then processed and padded by the NS2Collator according to the specified batch size. The NS2Collator is used as the collate_fn parameter of the DataLoader in ns2_trainer.py. I am not sure how the DataLoader processes the data output by the NS2Collator and why the DataLoader output is inconsistent with the NS2Collator output.