Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add batch_arg_name to all calls to _adjust_batch_size #4812

Merged
merged 1 commit into from
Nov 23, 2020

Conversation

M-Salti
Copy link
Contributor

@M-Salti M-Salti commented Nov 22, 2020

This PR fixes a bug in scale_batch_size when using a batch_arg_name other than the default 'batch_size'; not all the calls to _adjust_batch_size are passing batch_arg_name causing an exception when using batch_arg_name='bs' for example.

Here are two Colabs showing the bug and the fix:
1- (Bug) Install PL from the master branch: https://colab.research.google.com/drive/1VgLVfBl2xB0TJz4Js_xDSgz-Ae_fi_Jt
2- (Fix) Install PL from my master branch: https://colab.research.google.com/drive/1yierw80jgJb0qukBiCw54iNSelK8qJYB
The only difference is in the first cell (installation source).

Add `batch_arg_name` to all calls to `_adjust_batch_size`
@codecov
Copy link

codecov bot commented Nov 22, 2020

Codecov Report

Merging #4812 (2fb4f6c) into master (8601268) will not change coverage.
The diff coverage is 50%.

@@          Coverage Diff           @@
##           master   #4812   +/-   ##
======================================
  Coverage      93%     93%           
======================================
  Files         117     117           
  Lines        9010    9010           
======================================
  Hits         8383    8383           
  Misses        627     627           

@rohitgr7 rohitgr7 added bug Something isn't working tuner labels Nov 22, 2020
@rohitgr7 rohitgr7 added this to the 1.0.x milestone Nov 22, 2020
@rohitgr7 rohitgr7 merged commit cd90dd4 into Lightning-AI:master Nov 23, 2020
Borda pushed a commit that referenced this pull request Nov 23, 2020
Add `batch_arg_name` to all calls to `_adjust_batch_size`

(cherry picked from commit cd90dd4)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working tuner
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants