Code works with SourceImageNet.java (the initial code) proving that it work with a tiny dataset (1200 training chihuahuas and macaws plus 20 testing)
git add BaseForImageNet.java