I have this problem althoght I reduced my batch size of dataset
RuntimeError: [enforce fail at …\c10\core\CPUAllocator.cpp:79] data. DefaultCPUAllocator: not enough memory: you tried to allocate 754974720 bytes.