我正在做Pytorch到MindSpore的迁移,是在现有Pytorch代码的基础上进行迁移的,想知道如何把这个代码迁移到Mindspore上,并在epoch中训练。
原本代码数据集定义如下:
****************************************************解答*****************************************************
Dataloader都有对应的:mindspore.dataset.GeneratorDataset — MindSpore master 文档