pypi pytorch-lightning 0.4.9
Various ddt improvements

latest releases: 2.4.0, 2.3.3, 2.3.2...
5 years ago

This release does the following:

  • Moves SLURM resubmit from test-tube to PL (which removes the need for cluster parameter).
  • Cluster checkpoint done by Lightning now (not test-tube). Also doesn't require a checkpoint object to restore weights when on cluster.
  • Loads all models on CPU when restoring weights to avoid OOM issues in PyTorch. User now needs to move to GPU manually. However, if using Lightning, lightning will move to correct GPUs automatically.
  • Fixes various subtle bugs in DDP implementation.
  • documentation updates

Don't miss a new pytorch-lightning release

NewReleases is sending notifications on new releases.