when I train a model,
the input code is like the follows:
head = torch.from_numpy(head).to(self.device)
the loss code is like:
loss_sum += loss.item()
but I got a “cuda out of memory” after 8784 iterators
why?