pere's picture
new pytorch model
78ab986