Skip to main content

A simple trainer class for pytorch with tensorboard support

Project description

Base class for training pytorch models

Usage:

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."" - http://karpathy.github.io/2019/04/25/recipe/

Continue training a model after stopping at an epoch

Metrics

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 details)

Uploaded Source

File details

Details for the file pytorch-trainer-tensorboard-0.4.9.tar.gz.

File metadata

  • Download URL: pytorch-trainer-tensorboard-0.4.9.tar.gz
  • Upload date:
  • Size: 8.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.0 importlib_metadata/3.10.1 pkginfo/1.8.2 requests/2.22.0 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.3

File hashes

Hashes for pytorch-trainer-tensorboard-0.4.9.tar.gz
Algorithm Hash digest
SHA256 b5a8e8036a8cc7782b4c9594e80ee2b8e0cd09725617741d1bdf04b93dbadc1c
MD5 eca74c9cde09d546d4a8fe925a76be6f
BLAKE2b-256 6e97f5e8b82e944005173d8ab2bfca8cf35f58501eb5b2a0acd92144dd684553

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page