pytorch_tao.plugins.TrainPlugin

class pytorch_tao.plugins.TrainPlugin

Plugin for training phase. For more detail see BasePlugin

__init__()

Methods

__init__()
after_use()
attach(engine)
set_engine(engine)

Attributes