Hi,
You can remove data, label= Variable(data), Variable(label), this is not needed anymore.
Have you tried using a batch size of 1. It looks like your model is quite large.
Also you can run with anomaly mode enabled to see which part of the backward tries to allocate this 7GB Tensor.