jpmorganchase / llm-email-spam-detection Goto Github PK

OutOfMemoryError                          Traceback (most recent call last)

in ()

17

18     # Train LLMs

---> 19     train_llms(

20         list(range(5)),

21         ["ling", "sms", "spamassassin", "enron"],


19 frames

/ usr/local/lib/python3.10 / dist - packages/transformers/activations. Py in forward (self, input)

55

56     def forward(self, input: Tensor) -> Tensor:

---> 57         return self.act(input)

58

59



OutOfMemoryError: CUDA out of memory. Tried to allocate 72.00 MiB (GPU 0; 15.77 GiB total capacity; 14.60 GiB already allocated; 46.12 MiB free; 14.75 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