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

模組內容