The main reason for this is that DataLoader
the 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.
Solve the problem that the data generated by DataLoader is all zero
Guess you like
Origin blog.csdn.net/qq_29391809/article/details/124201278
Recommended
Ranking