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

Fix inconsistency on passed number of samples in batch-then-sample mode #515

Merged
merged 14 commits into from
Jun 20, 2024
Prev Previous commit
Next Next commit
fix unit test
  • Loading branch information
XianzheMa committed Jun 17, 2024
commit cb5a4c99b25d0c56d526fb9e7f8a35115104b4ad
Original file line number Diff line number Diff line change
Expand Up @@ -943,7 +943,7 @@ def mock_forward(data):

trainer.train()

assert trainer._num_samples == expected_bts_size * num_batches
assert trainer._num_samples == batch_size * num_batches
assert test_on_batch_begin.call_count == len(trainer._callbacks) * num_batches
assert test_on_batch_end.call_count == len(trainer._callbacks) * num_batches
assert test_downsample_batch.call_count == num_batches
Expand Down
Loading