Attentive-Group-Recommendation
Attentive-Group-Recommendation copied to clipboard
您好,在运行代码时遇到了内存溢出的问题。
运行错误如下: RuntimeError: [enforce fail at ..\c10\core\CPUAllocator.cpp:72] data. DefaultCPUAllocator: not enough memory: you tried to allocate 986880 bytes. Buy new RAM!
----》错误指向这一行 training(agree, dataset.get_group_dataloader(config.batch_size), epoch, config, 'group'),请问这个可以通过减小batch_size来处理吗? 谢谢