Autograd expecting wrong variable version

I think you are running into a similar issue as described here.
In particular:

One source of error is that the model needs the unmodified data_in but the versioning does not capture regions of data_in.

and you could fix it by using a = FC2(b_storage[t].clone()).

1 Like