add DataParallel import from torch.nn

#6

sorry this is rather rudimentary suggestion

NVIDIA org

Thank you for the suggestions. We updated the model card adding "from torch.nn import DataParallel".

nada5 changed pull request status to closed

Sign up or log in to comment