Skip to content

Instantly share code, notes, and snippets.

@securetorobert
Created October 1, 2020 16:19
Show Gist options
  • Save securetorobert/ea377777ddf4eefb7f5d9c4908c6d754 to your computer and use it in GitHub Desktop.
Save securetorobert/ea377777ddf4eefb7f5d9c4908c6d754 to your computer and use it in GitHub Desktop.
Train a model
optimizer = optimizers.SGD(learning_rate=0.01, momentum=0.9, decay=0.0005)
model.compile(optimizer=optimizer, loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(ds, epochs=1, steps_per_epoch=2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment