spectre/utils/__init__.py

from ._utils import (
    fix_random_seeds, 
    to_ntuple,
    to_1tuple,
    to_2tuple,
    to_3tuple,
    to_4tuple,
)
from .modeling import (
    deactivate_requires_grad_and_to_eval,
    activate_requires_grad_and_to_train,
    update_momentum,
    update_drop_path_rate,
    repeat_token,
    expand_index_like,
    get_at_index,
    set_at_index,
    mask_at_index,
    mask_bool,
    patchify,
    random_token_mask,
    resample_abs_pos_embed,
    resample_abs_pos_embed_nhwdc,
    resample_patch_embed,
    feature_take_indices,
    global_pool_nlc,
    cat_keep_shapes,
    uncat_with_shapes,
    last_token_pool,
    Format,
    nchwd_to,
    nhwdc_to,
)

__all__ = [
    "fix_random_seeds",
    "to_ntuple",
    "to_1tuple",
    "to_2tuple",
    "to_3tuple",
    "to_4tuple",
    "deactivate_requires_grad_and_to_eval",
    "activate_requires_grad_and_to_train",
    "update_momentum",
    "update_drop_path_rate",
    "repeat_token",
    "expand_index_like",
    "get_at_index",
    "set_at_index",
    "mask_at_index",
    "mask_bool",
    "patchify",
    "random_token_mask",
    "resample_abs_pos_embed",
    "resample_abs_pos_embed_nhwdc",
    "resample_patch_embed",
    "feature_take_indices",
    "global_pool_nlc",
    "cat_keep_shapes",
    "uncat_with_shapes",
    "last_token_pool",
    "Format",
    "nchwd_to",
    "nhwdc_to",
]