Java Deep Learning Cookbook

How it works...

In step 1, the value of saveUpdater is going to be true if we plan to train the model at a later point. We have also discussed pre-trained models provided by DL4J's model zoo API. Once we add the dependency for deeplearning4j-zoo, as mentioned in step 1, we can load pre-trained models such as VGG16, as follows:

ZooModel zooModel = VGG16.builder().build(); ComputationGraph pretrainedNet = (ComputationGraph)    zooModel.initPretrained(PretrainedType.IMAGENET);

DL4J has support for many more pre-trained models under its transfer learning API.

Fine-tuning a configuration is the process of taking a model that was trained to perform a task and training it to perform another similar task. Fine-tuning configurations is specific to transfer ...