Skip to content

Instantly share code, notes, and snippets.

@eileen-code4fun

eileen-code4fun/cifar10_viz.py

Created May 28, 2021 02:08
Show Gist options
  • Save eileen-code4fun/0ca86a0ca860dac9ce41f78e12c4b779 to your computer and use it in GitHub Desktop.
Save eileen-code4fun/0ca86a0ca860dac9ce41f78e12c4b779 to your computer and use it in GitHub Desktop.
Download ZIP
CIFAR10 Viz
Raw
cifar10_viz.py
import os
tensorboard_callback = tf.keras.callbacks.TensorBoard(
log_dir=os.environ['AIP_TENSORBOARD_LOG_DIR'],
histogram_freq=1
)
model.fit(train_dataset, epochs=args.epochs, validation_data=val_dataset, callbacks=[tensorboard_callback])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment