I’m taking a look through this and it occurred that you may want to use at.set_subtensor(param, value) to update variables in the graph. That said, this gives me a shape error that I’m still trying to figure out.
I’m taking a look through this and it occurred that you may want to use at.set_subtensor(param, value) to update variables in the graph. That said, this gives me a shape error that I’m still trying to figure out.