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 實例，而不是這個函式，因為前者會處理執行已註冊的 hook，而後者會靜默地忽略它們。

training: bool

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

模組內容