Fix issue with `torchvision`'s `ImageNet`

[dedeswim]

When instantiating torchvision's ImageNet or ImageFolder, the download argument is passed, even if these two classes do not take this argument.

This PR removes the argument from the torch_kwargs dict before the classes are isntantiated.