Closed alfrem closed 4 years ago
I suppose this can be done in model_zoo.py, right?
yes :) just follow the example of the vgg19 or so. Let me know if it worked!
ResNet is now implemented: c3fc8b6 (for this to work you need keras applications >=1.0.7; this is installed by default when you install keras v >=2.2.4) The rest of AIDeveloper will now be adjusted to work with keras v. 2.2.4
SqueezeNet could be taken from here: https://github.com/rcmalli/keras-squeezenet
SqueezeNet is now added to the model_zoo: 3662af5
ResNet v2 is not working: Error: module 'keras.applications' has no attribute 'resnet'
Solved via 80696fc
Just as an idea. Would be interesting to have AlexNet, ResNet or SqueezeNet.
Resnet can be tricky with parallelism and residual module but provide excelent results.
Thank you.