Fix another reference to torch package
Fix another reference to torch package

Deepomatic/DIGITS,ethantang95/DIGITS,TimZaman/DIGITS,Lucaszw/DIGITS,TimZaman/DIGITS,batra-mlp-lab/DIGITS,Deepomatic/DIGITS,brainstorm-ai/DIGITS,Deepomatic/DIGITS,batra-mlp-lab/DIGITS,TimZaman/DIGITS,gheinrich/DIGITS-GAN,ethantang95/DIGITS,dongjoon-hyun/DIGITS,ethantang95/DIGITS,batra-mlp-lab/DIGITS,brainstorm-ai/DIGITS,gheinrich/DIGITS-GAN,Lucaszw/DIGITS,batra-mlp-lab/DIGITS,Lucaszw/DIGITS,gheinrich/DIGITS-GAN,ethantang95/DIGITS-GAN,dongjoon-hyun/DIGITS,ethantang95/DIGITS-GAN,Deepomatic/DIGITS,ethantang95/DIGITS-GAN,jmancewicz/DIGITS,bygreencn/DIGITS,dongjoon-hyun/DIGITS,bygreencn/DIGITS,ethantang95/DIGITS-GAN,bygreencn/DIGITS,jmancewicz/DIGITS,TimZaman/DIGITS,bygreencn/DIGITS,ethantang95/DIGITS,Lucaszw/DIGITS,gheinrich/DIGITS-GAN,jmancewicz/DIGITS,jmancewicz/DIGITS,brainstorm-ai/DIGITS,dongjoon-hyun/DIGITS,brainstorm-ai/DIGITS