6.6Gb is a big one. During training, because of additional space required for graph, it’ll need much more than 11gb you have.
I would try to call model.float() before torchinfo() in case model is using float64.