CUDA out of memory. Tried to allocate 3.82 GiB (GPU 0; 14.73 GiB total capacity; 12.46 GiB already allocated; 7.88 MiB free; 13.79 GiB)

Question

epochs = 20

for epoch in range(0, epochs + 1):
    # Training
    if epoch > 0:  # test untrained net first
        model.train()
        train_loss = 0