final_model.pt 43 MB