Cuda oom when calculating loss function

[my3bikaht (Sergey)]

@Bika

one_hot = torch.zeros_like(pred).scatter(1, gold.view(-1, 1), 1)

Error says that creating this tensor requires 5gb of GPU memory.
What are the shapes of pred and gold?