Thank you for your quick response.
When training step is executed, it consumes less than 8 GB. But when it goes into validation, at “model.test()” step it throws error. And particular in testing code, it is in “self.LR_est = self.netG2(self.SR)” line.
Currently I stopped using two models in validation and now it is working(with only one model), but using two models it give me error.
In test method, it iteratively consumes more and more memory until all 16GB and then throwing error.
In testing, I am using batch size of 1 only.