Computation of SSIM takes so much GPU memory?

[franciszchen]

When computing the ssim of the output of a CNN (batch size 8, image resolution is 192*192), the consumed GPU memory are doubled. It seems that the ssim should require little GPU memory. I am curious about this problem. Thx~

[shonesaji11]

For my project, i needed the ssim function. So by adding the line below at the start of the ssim function definition (in the init.py file) , i was able to reduce the memory consumption
torch.cuda.empty_cache()