Determine batch size from `labels` instead of `inputs`

davidtvs / pytorch-lr-finder

A learning rate range test implementation in PyTorch

MIT License

921 stars 120 forks source link

Closed NaleRaphael closed 4 years ago

NaleRaphael commented 4 years ago

Fix issue #57, and related tests are added.

davidtvs commented 4 years ago

/flake8-lint

davidtvs commented 4 years ago

LGTM. Thanks!