pytorch - SageMaker Neo 自定义 model_fn 和 predict_fn
问题描述
我正在查看 PyTorch SageMaker Neo 编译示例。他们似乎将 requiredneo_preprocess和neo_postprocess函数放在train.py文件中，而不是 aserve.py也包含model_fnand predict_fn。
我想知道，是否可以在使用 Neo 编译的同时使用serve.py我自己的model_fnand进行自定义？predict_fn
解决方案
neo_preprocess并且neo_postprocess是必需的，因为我们的容器将使用这两个函数。但是，您仍然可以将自己的serve.py与自己的一起使用model_fn，只需确保保留neo_preprocess和neo_postprocess.
推荐阅读
- ionic3 - ngx-mqtt TypeError: __WEBPACK_IMPORTED_MODULE_0__angular_core__.defineInjectable 不是函数
- ionic-framework - 在离子 4 中解散后打开加载
- php - 缓存 SQL 查找和检索数据
- html - 尝试将鼠标悬停在页脚菜单上时它会消失
- python - 在 Python 中实现多图
- python-3.x - Find the average of a value in a list of tuples within a list of dictionaries in Python
- vue.js - 我可以将 Laravel 与 Vue 店面模板混合使用吗
- java - 构造函数中的布尔值设置为 true
- angular - Angular 2 - 我的子路线没有打开任何东西
- swift - 在 Heroku 上使用 Vapor 3 配置 PostgreSQL 数据库