Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fixed Opacus's Runtime error with an empty batch (issue 612) (#631)
Summary: In case of an empty batch, in the ```clip_and_accumulate``` function, the ```per_sample_clip_factor``` variable is set to a tensor of size 0. However, the device was not specified, which throws a runtime error. Added it. Differential Revision: D53733081
- Loading branch information