Я смотрю на примеры компиляции PyTorch SageMaker Neo.Кажется, они помещают необходимые функции neo_preprocess и neo_postprocess в файл train.py, а не serve.py, который также включает model_fn и predict_fn.
Мне было интересно, можно ли вместо этого иметь пользовательские serve.py с моими собственными model_fn и predict_fn, при этом используя компиляцию Neo?