lecun-team | Master-IASD | boldeagle | Error | 0 | 100 | 0 | bytes: b'Traceback (most recent call last):\n File "/home/lamsade/testplatform/test-platform-a1/repos/Master-IASD/lecun-team/generate.py", line 39, in \n model.make_output(batch_size = 65536, tqdm_on = False)\n File "/home/lamsade/testplatform/test-platform-a1/repos/Master-IASD/lecun-team/algorithms/folded_deep_matrix_factorization_imprv.py", line 308, in make_output\n test_users_explicit_ds = self.R[test_users_idx].float() / 5\n ~~~~~~^^^^^^^^^^^^^^^^\ntorch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.43 GiB (GPU 0; 10.91 GiB total capacity; 662.84 MiB already allocated; 1.32 GiB free; 3.04 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF\n' |