Thanks for clarification!
Hopefully, in my case, the backprop operation is quite simple. I have implemented an customized backprop operation to solve the memory issue.
Thanks for clarification!
Hopefully, in my case, the backprop operation is quite simple. I have implemented an customized backprop operation to solve the memory issue.