add DataParallel import from torch.nn

#6
by yxzwayne - opened

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
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment