Fix RandAugment and TrivialAugment bugs (#4370)
Fix RandAugment and TrivialAugment bugs (#4370)

* Fix RA bugs.

* Fix bins for TA.
pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision