-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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
attempting to remove some speed issues #1482
Conversation
Codecov Report
@@ Coverage Diff @@
## master #1482 +/- ##
======================================
- Coverage 91% 91% -0%
======================================
Files 67 67
Lines 3745 3742 -3
======================================
- Hits 3403 3400 -3
Misses 342 342 |
benchmarks/test_trainer_parity.py
Outdated
assert_speed_parity(pl_times, pl_times, num_epochs) | ||
|
||
|
||
def assert_speed_parity(pl_times, pt_times, num_epochs): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
do we really need this two times? Can't we just import the other one?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fair, i was just trying to find the threshold number before finalizing pr. will refactor this
@@ -112,10 +112,6 @@ def process_output(self, output, train=False): | |||
num_gpus = self.num_gpus | |||
callback_metrics = self.reduce_distributed_output(callback_metrics, num_gpus) | |||
|
|||
for k, v in callback_metrics.items(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
why remove this?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it’s done twice. this is the first time but it’s done again at the end of the method
Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com>
Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com>
* removed some .items * added speed tests * added speed tests * Update benchmarks/test_rnn_parity.py Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * Update benchmarks/test_trainer_parity.py Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * fix lost model reference * added speed tests Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
trying to match exact PT speed