Addresses issue #145 as per @fmessa's suggestion. (#527)
Addresses issue #145 as per @fmessa's suggestion. (#527)

* Addresses issue #145 as per @fmessa's suggestion.

* Removed blank line for styling.

pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision