fine-tune on custom dataset
#1
by
asifdegr8
- opened
I want to finetune this model using my own custom dataset for the purpose of generating synthetic images. Would you please help how to achieve this?
This model was trained on CIFAR 10 dataset
requirements.txt
tensorflow-cpu
Flask==3.0.0
keras==3.4.0
numpy==1.26.4
matplotlib==3.7.1
Flask-Cors==4.0.1
huggingface_hub==0.23.5
gunicorn
python-dotenv==1.0.1
Kaggle Link (Model Training):
https://www.kaggle.com/code/pratapkr/conditional-gan-cifar10
GitHub:
https://github.com/pratap-nitjsr/Random-Image-Generator