Include batch_size as an argument, instead of sample_rate
Summary: Other examples in Opacus follow this logic for arguments related to batch size:
- batch_size and batch_size_test
CIFAR-10 still uses the sample_rate * len(train_data).
This diff updates that, to aslo make it clearer (someone was confused here)
Differential Revision: D38799564
This pull request was exported from Phabricator. Differential Revision: D38799564
This pull request was exported from Phabricator. Differential Revision: D38799564
This pull request was exported from Phabricator. Differential Revision: D38799564
This pull request was exported from Phabricator. Differential Revision: D38799564
This pull request was exported from Phabricator. Differential Revision: D38799564