Issue with classification-train on a custom dataset

[MasoumehSharafi]

Hello,
I want to train the model on a different dataset, but I cannot save the models for adaptation. Could you explain how to train and save the pre-trained models like your saved ones in the models folder?
Thanks.