search

bhky / opennsfw2

Keras implementation of the Yahoo Open-NSFW model
MIT License
361 stars 47 forks source link

How to continue train the model? #14

Closed lukywangwang closed 1 year ago

lukywangwang commented 1 year ago

I have my own porn dataset. How can I resume the training process with the the current model? thanks Best.

bhky commented 1 year ago

Hi! Thanks for the message.

The model object is a normal Keras model. You can follow this section on how to do the data preprocessing and instantiate the model. Then you can just use the standard way of training a TensorFlow/Keras model with your dataset.

If you are new to using TensorFlow/Keras, you can take a look at this guide.