Skip to content

Instantly share code, notes, and snippets.

@trongan93

trongan93/ship_detection_training

Last active June 12, 2020 08:47
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save trongan93/cef76a5f0b0f9dc2b64070b894ec6401 to your computer and use it in GitHub Desktop.
Save trongan93/cef76a5f0b0f9dc2b64070b894ec6401 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ship_detection_training
# optimization setup
sgd = SGD(lr=0.01, momentum=0.9, nesterov=True)
model.compile(
loss='categorical_crossentropy',
optimizer=sgd,
metrics=['accuracy'])
# training
model.fit(
X_train,
y_train,
batch_size=32,
epochs=18,
validation_split=0.2,
shuffle=True,
verbose=2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment