Skip to content

Instantly share code, notes, and snippets.

@eileen-code4fun

eileen-code4fun/cifar10_model_train.py

Created May 26, 2021 03:03
Show Gist options
  • Save eileen-code4fun/a20635e6629c3c45a4a5c840b7b5bb84 to your computer and use it in GitHub Desktop.
Save eileen-code4fun/a20635e6629c3c45a4a5c840b7b5bb84 to your computer and use it in GitHub Desktop.
Download ZIP
CIFAR10 Model Train
Raw
cifar10_model_train.py
model.fit(train_dataset, epochs=5, validation_data=val_dataset)
model.evaluate(test_dataset, verbose=2)
# Output
# loss: 1.0985 - accuracy: 0.6060
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment