-
Notifications
You must be signed in to change notification settings - Fork 111
Is it possible to segment the definition of the function to be taken the gradient? #118
Comments
Conditionals are allowed in non-optimized mode (which is the default) On Mon, May 9, 2016 at 5:20 PM David Lopes Macedo [email protected]
|
Dear friends, Could someone tells me what is the error bellow:
I am using cudnn and getting this error:
When I use the following code, it works great:
|
Is |
Try |
I am using cuddn... Input and output are tensors... The following code is ok:
But I still don't understand what is wrong with the code bellow:
Full logs: In 3 module of nn.Sequential: WARNING: If you see a stack trace below, it doesn't point to the place where this error occured. Please use only the one above. |
Does it work on plain NN, on the CPU?
|
No. When trying to run plain NN on the CPU with tensor of type float (no cuda or cunn), I got the following error:
|
Dear friend,
Is it possible to segment the definition of the function to be taken the gradient?
For example:
Thanks,
David
The text was updated successfully, but these errors were encountered: