mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
81ca323751
fixed head to gap->norm->fc as per convnext, along with option for norm->gap->fc failed tests due to clip convnext models, davit tests passed
16 lines
764 B
Python
16 lines
764 B
Python
from .agc import adaptive_clip_grad
|
|
from .checkpoint_saver import CheckpointSaver
|
|
from .clip_grad import dispatch_clip_grad
|
|
from .cuda import ApexScaler, NativeScaler
|
|
from .decay_batch import decay_batch_step, check_batch_size_retry
|
|
from .distributed import distribute_bn, reduce_tensor, init_distributed_device,\
|
|
world_info_from_env, is_distributed_env, is_primary
|
|
from .jit import set_jit_legacy, set_jit_fuser
|
|
from .log import setup_default_logging, FormatterNoInfo
|
|
from .metrics import AverageMeter, accuracy
|
|
from .misc import natural_key, add_bool_arg, ParseKwargs
|
|
from .model import unwrap_model, get_state_dict, freeze, unfreeze
|
|
from .model_ema import ModelEma, ModelEmaV2
|
|
from .random import random_seed
|
|
from .summary import update_summary, get_outdir
|