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

模块内容