Java Deep Learning Cookbook

How to do it...

1. Use FineTuneConfiguration() to manage modifications in the model configuration:

FineTuneConfiguration fineTuneConf = new FineTuneConfiguration.Builder() .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT) .updater(new Nesterovs(5e-5)) .activation(Activation.RELU6) .biasInit(0.001) .dropOut(0.85) .gradientNormalization(GradientNormalization.RenormalizeL2PerLayer) .l2(0.0001) .weightInit(WeightInit.DISTRIBUTION) .seed(seed) .build();

2. Call fineTuneConfiguration() to fine-tune the model configuration:

MultiLayerNetwork newModel = new TransferLearning.Builder(oldModel).fineTuneConfiguration(fineTuneConf).build();