Skip to content

Instantly share code, notes, and snippets.

@pabloformoso

pabloformoso/VGG16_Keras_OutOfTheBox.py

Created April 26, 2018 12:51
Show Gist options
  • Save pabloformoso/263454564dbfbcf560da30aec68766d1 to your computer and use it in GitHub Desktop.
Save pabloformoso/263454564dbfbcf560da30aec68766d1 to your computer and use it in GitHub Desktop.
Download ZIP
Built in VGG16 network with Keras
Raw
VGG16_Keras_OutOfTheBox.py
from keras.optimizers import SGD
from keras.applications.vgg16 import VGG16
model = VGG16(weight='imagenet', include_top=True)
sgd = SGD(lr=0.1, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(optimizer=sgd, loss='categorical)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment