Skip to content
This repository has been archived by the owner on Sep 27, 2020. It is now read-only.

Error in backward #2

Closed
xgniu opened this issue Jan 16, 2018 · 1 comment
Closed

Error in backward #2

xgniu opened this issue Jan 16, 2018 · 1 comment

Comments

@xgniu
Copy link

xgniu commented Jan 16, 2018

Thanks for your implementation of conv-lstm. However, there may exist some bugs in the code. I use the code as a part of my project. The convolutional features are extracted from images and are passed to the conv-lstm. Following the conv-lstm are a fully-connected layer and the loss. But the loss.backward() will report an error, which tells that the 'retain_graph' parameter should be true. However, setting retain_graph=True will consume more and more memory and slow down the program.

@automan000
Copy link
Owner

Hello, I cannot reproduce this bug. Could you provide the certain part of your code?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants