mmengine/docs/resources/config/my_module.py

7 lines
101 B
Python

from mmengine.registry import OPTIMIZERS
@OPTIMIZERS.register_module()
class CustomOptim:
pass