I ran the test and added (out.volatile=False and out.requires_grad=True) and it worked fine.
the results are: the GPUs RAM not increasing so is the CPU RAM is not increasing.
I also added the print for stage1 and the grad is None and my RAM on CPU keep increasing.