Skip to content
Discussion options

You must be logged in to vote

The issue is that you are never releasing the computation graph.

loss_total = loss_total + float(loss)

should fix this.

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Answer selected by DogitoErgoSum
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants