Besides the parameters and inputs intermediate forward activations could allocate a lot of memory as explained in e.g. this post.
1 Like
Besides the parameters and inputs intermediate forward activations could allocate a lot of memory as explained in e.g. this post.