ts.torch_handler.unit_tests.models 包

子模块

ts.torch_handler.unit_tests.models.base_model 模块

仅用于测试 BaseHandler 的简单前馈模型

class ts.torch_handler.unit_tests.models.base_model.ArgmaxModel(*args, **kwargs)

基类：Module

forward(*input)

定义每次调用时执行的计算。

所有子类都应覆盖此方法。

注意

虽然前向传递的步骤需要在此函数中定义，但应随后调用 Module 实例，而不是此函数，因为前者负责运行已注册的钩子，而后者会静默地忽略它们。

training: bool

ts.torch_handler.unit_tests.models.base_model.save_pt_file(filepath='base_model.pt')

模块内容