Hi, I happen to be using the Efficient model, so let me answer that question
I’ll tell you a rule for computing video memory.

WechatIMG3720×421 98.9 KB

efficientnet-b0  parms is 5.3M.
For details, please see https://github.com/Lornatang/EfficientNet/blob/master/README.md