Hi, I am trying to run the ResNet18 model using GTX 1080 GPU but the Cuda out-of-memory error arises. I reduced the baches size to 4 and use
    learn=0
    gc.collect()
    torch.cuda.empty_cache()

but they didn’t work.
could anyone give me some advice?