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())
.