mirror of
https://github.com/aigc3d/LAM_Audio2Expression.git
synced 2026-02-04 09:29:24 +08:00
30 lines
486 B
Python
30 lines
486 B
Python
"""
|
|
The code is base on https://github.com/Pointcept/Pointcept
|
|
"""
|
|
|
|
|
|
class HookBase:
|
|
"""
|
|
Base class for hooks that can be registered with :class:`TrainerBase`.
|
|
"""
|
|
|
|
trainer = None # A weak reference to the trainer object.
|
|
|
|
def before_train(self):
|
|
pass
|
|
|
|
def before_epoch(self):
|
|
pass
|
|
|
|
def before_step(self):
|
|
pass
|
|
|
|
def after_step(self):
|
|
pass
|
|
|
|
def after_epoch(self):
|
|
pass
|
|
|
|
def after_train(self):
|
|
pass
|