saved_model/model.pth