for example:
self.cnn = model.densenet121(pretrained=True).features
x1 = self.cnn(x1)
x2 = self.cnn(x2)
x3 = self.cnn(x3)
got cuda oof in pytorch 1.0, the memory increases on each line, which, I think, should not.
The similar code run in 0.4.1 is ok. Not know why?
thanks!