小编Jai*_*ata的帖子

TorchServe MAR 每个模型多个 Python 文件

有没有办法在调用 MAR 之前将模型拆分为多个文件?

例如,如果我有文件base_model.py

import torch
from torch import nn


class BaseModel(nn.Module):
    ...
Run Code Online (Sandbox Code Playgroud)

以及继承自的模型BaseModel

from base_model import BaseModel


class FullModel(BaseModel):
    ...
Run Code Online (Sandbox Code Playgroud)

对于处理程序来说也是如此。我不知道如何在多个文件之间拆分代码。

python pytorch

5
推荐指数
1
解决办法
2515
查看次数

标签 统计

python ×1

pytorch ×1