/swift/megatron/callbacks/__init__.py

# Copyright (c) ModelScope Contributors. All rights reserved.
from .base import MegatronCallback
from .mapping import megatron_callbacks_map