Skip to content
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
dataHelpers Eclipse Migration Initial Commit Sep 7, 2019
EditAtBottleneckOthersFrozen.java Eclipse Migration Initial Commit Sep 7, 2019
EditLastLayerOthersFrozen.java
FineTuneFromBlockFour.java Eclipse Migration Initial Commit Sep 7, 2019
FitFromFeaturized.java Eclipse Migration Initial Commit Sep 7, 2019
README.md Eclipse Migration Initial Commit Sep 7, 2019

README.md

Demonstrates use of the dl4j transfer learning API which allows users to

- construct a model based off an existing model
- "freeze" certain layers
- fine tune learning parameters

Explore preferably in the following order.

  1. Read TransferLearning.md
  2. Run "EditLastLayerOthersFrozen" to modify just the last layer in org.deeplearning4j.transferlearning.vgg16 and use it to fit the dataset. This is expected to run a while depending on your hardware.
  3. Build the same architecture in (2) but with featurized datasets
    • Run "FeaturizedPreSave" which will featurize ~3000 images by passing them through org.deeplearning4j.transferlearning.vgg16. This is also expected to run a while depending on your hardware.
    • Run "FitFromFeaturize" which will show you how to fit to presaved data so you can iterate quicker with different learning parameters. Fitting with the presaved dataset is very quick.
  4. "EditAtBottleneckOthersFrozen" for a look into how to rework model architecure by adding/removing vertices
  5. "FineTuneFromBlockFour" to show you how to continue training on a saved transfer learning model

You can read the documentation for the Transfer Learning API at https://deeplearning4j.org/transfer-learning.

You can’t perform that action at this time.