Skip to main content

A simple trainer class for pytorch with tensorboard support

Project description

Base class for training pytorch models


Launch tensorboard:

Visualize prediction dynamics:

""visualize prediction dynamics. I like to visualize model predictions on a fixed test batch during the course of training. The “dynamics” of how these predictions move will give you incredibly good intuition for how the training progresses. Many times it is possible to feel the network “struggle” to fit your data if it wiggles too much in some way, revealing instabilities. Very low or very high learning rates are also easily noticeable in the amount of jitter."" -

Continue training a model after stopping at an epoch


sanity checks:

  1. check loss reduces on a fixed data samples (say 1 or 2 inputs) by iterating

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytorch-trainer-tensorboard-0.4.9.tar.gz (8.7 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page