search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.59k stars 510 forks source link

Increased model size #1384

Closed RadouaneK closed 1 year ago

RadouaneK commented 1 year ago

💡 Your Question

I trained a yolo_nas_s model on my custom datasets, but I have noticed the the model size has increased after training, from 73MB to 256 MB. is it normal ? the input size I used is: 256. and the number of classes is: 40

BloodAxe commented 1 year ago

The checkpoint may contain optimizer state (useful if you want to resume training) and also ema model state dict as well, so yet it is normal.