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

Training YOLONAS from scratch for using it in a commercial application #901

Closed albertofernandezvillan closed 1 year ago

albertofernandezvillan commented 1 year ago

As far as I understand for the license, if I used pre-trained weights for training YOLONAS (fine-tune my model on my dataset), I cannot use it in commercial applications. Is it right?

If so, when training:

from super_gradients.training import models
model = models.get('yolo_nas_l', 
                   num_classes=len(dataset_params['classes']), 
                   pretrained_weights="coco"
                   )

how to change this and train it from scratch?

dagshub[bot] commented 1 year ago

Join the discussion on DagsHub!

NatanBagrov commented 1 year ago

Hello, to get a model without pre-trained weights, you should remove the relevant part (pretrained_weights="coco"). This is applicable to any model.

For more information regarding the API, please read the docs, here is the relevant part

albertofernandezvillan commented 1 year ago

Thanks, and what about "I used pre-trained weights for training YOLONAS (fine-tune my model on my dataset), I cannot use it in commercial applications. Is it right?"

shaydeci commented 1 year ago

@albertofernandezvillan You are right.

NeovMs commented 1 year ago

If we use the pre-trained models for knowledge distillation to pre-train the Yolo-NAS from scratch, we're not strictly using the weights. In this case, can we get round the licence ?