model_final.pth