Skip to content

Instantly share code, notes, and snippets.

@iavinas
Created May 24, 2020 11:07
Show Gist options
  • Save iavinas/51721faaa26d9d473688beaa25d84df3 to your computer and use it in GitHub Desktop.
Save iavinas/51721faaa26d9d473688beaa25d84df3 to your computer and use it in GitHub Desktop.
Resnet50_model.compile(loss='categorical_crossentropy', optimizer='rmsprop', metrics=['accuracy'])
checkpointer = ModelCheckpoint(filepath='saved_models/weights.best.Resnet50.hdf5',
verbose=1, save_best_only=True)
Resnet50_model.fit(train_Resnet50, train_targets,
validation_data=(valid_Resnet50, valid_targets),
epochs=20, batch_size=20, callbacks=[checkpointer], verbose=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment