Welcome to Tez’s documentation!
Tez is a simple pytorch trainer to make your life easy. It comes with some useful dataset classes and callbacks.
from tez import Tez, TezConfig class MyModel(nn.Module): def __init__(self): super().__init__() # do something here def optimizer_scheduler(self): # return optimizer and scheduler here return opt, sch def forward(self, arg1, arg2): # do something here return outputs, loss, metrics
In Tez, the dataset class and model’s forward function are closely related. The output names from dataset class must be same as the input arguments in forward function of the model.