pere's picture
converted to pytorch
33aed67