Skip to content

Instantly share code, notes, and snippets.

@alankbi
Last active February 16, 2020 01:33
Show Gist options
  • Save alankbi/6d0ccb47bbf13d741c9e40f0e6d8ca23 to your computer and use it in GitHub Desktop.
Save alankbi/6d0ccb47bbf13d741c9e40f0e6d8ca23 to your computer and use it in GitHub Desktop.
Advanced training with Detecto
import matplotlib.pyplot as plt
val_dataset = core.Dataset('validation_images/')
losses = model.fit(loader, val_dataset, epochs=10, learning_rate=0.001,
lr_step_size=5, verbose=True)
plt.plot(losses)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment