Solve the problem that the data generated by DataLoader is all zero

The main reason for this is that DataLoaderthe data in the GPU cannot be processed and can only be processed by the CPU. Therefore, the return value of the function in the Dataset __getitem__can only be the data in the CPU.

Guess you like

Origin blog.csdn.net/qq_29391809/article/details/124201278