我正在看PyTorch SageMaker Neo编译示例。他们似乎将所需的neo_preprocess和neo_postprocess函数放置在train.py文件中，而不是同时包含serve.py和model_fn的{{1}}。
我想知道，是否有可能使用我自己的predict_fn和serve.py创建一个自定义的model_fn，同时仍使用Neo编译？
答案 0 :(得分：0)
neo_preprocess和neo_postprocess是必需的，因为我们的容器将使用这两个函数。但是，您仍然可以将自己的serve.py与自己的model_fn一起使用，并确保保留neo_preprocess和neo_postprocess.