ribesstefano's picture
Working loading and predicting with pretrained model + Added scaling in "pure" pytorch during forwarding
15216c3