Fixed details in torch model