Skip to content

Fix memory issue#233

Open
ZhipengXue97 wants to merge 1 commit intoyanx27:masterfrom
ZhipengXue97:master
Open

Fix memory issue#233
ZhipengXue97 wants to merge 1 commit intoyanx27:masterfrom
ZhipengXue97:master

Conversation

@ZhipengXue97
Copy link

Computations involving variables that require gradients will keep history. avoid using such variables in computations which will live beyond your training loops.

Reference to PyTorch FAQ:
"https://github.com/lisroach/pytorch/blob/53fceff1e1e08e6d11663d260e136eaa1a74be79/docs/source/notes/faq.rst"

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant