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')

模块内容