如何训练预训练 model (MNIST) tensorflow

Question

I have a project with Fashion MNIST, which predicts clothes from uploaded images, and I want to make some improvements with it. Is it possible to modify my project that it will train automatically after each uploaded and prediction?

Answer 1

You can train your model manually by using the transfer learning technique(Transfer learning is a method of reusing an already trained model for another task).

1. Instantiate a base model and load pre-trained weights into it.
2. Freeze all layers in the base model by setting trainable = False.
3. Create a new model on top of the output of one (or several) layers from the base model. Train your new model on your new dataset.

Please refer to this gist for working code example. Thank You.