chinmaydan's picture
changed requirement from torch to pytorch
bdb63c0