from cassle.my_method.cbam_distill_imagenet import base_cbam_wrapper2
from cassle.my_method.cbam_distill_1propose1 import base_cbam_wrapper5
from cassle.my_method.cbam_distill import base_cbam_wrapper4

__all__ = ["base_cbam_wrapper2", "base_cbam_wrapper5", "base_cbam_wrapper4"]

MY_METHODS = {
    "imagenet": base_cbam_wrapper2,
    "1p1": base_cbam_wrapper5,
    "1p12": base_cbam_wrapper4,
}