pytorch-image-models/timm/utils/__init__.py

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
Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00			`from .agc import adaptive_clip_grad`
Reorg of utils into separate modules 2020-09-07 13:56:00 -07:00			`from .checkpoint_saver import CheckpointSaver`
Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00			`from .clip_grad import dispatch_clip_grad`
Reorg of utils into separate modules 2020-09-07 13:56:00 -07:00			`from .cuda import ApexScaler, NativeScaler`
Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry... 2022-07-18 18:01:39 -07:00			`from .decay_batch import decay_batch_step, check_batch_size_retry`
Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed. 2022-09-23 16:08:59 -07:00			`from .distributed import distribute_bn, reduce_tensor, init_distributed_device,\`
			`world_info_from_env, is_distributed_env, is_primary`
Add --fuser arg to train/validate/benchmark scripts to select jit fuser type 2022-01-17 13:54:25 -08:00			`from .jit import set_jit_legacy, set_jit_fuser`
Reorg of utils into separate modules 2020-09-07 13:56:00 -07:00			`from .log import setup_default_logging, FormatterNoInfo`
			`from .metrics import AverageMeter, accuracy`
Davit update formatting and fix grad checkpointing (#7) 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 2023-01-15 14:34:56 -08:00			`from .misc import natural_key, add_bool_arg, ParseKwargs`
Small post-merge tweak for freeze/unfreeze, add to __init__ for utils 2021-10-06 17:00:27 -07:00			`from .model import unwrap_model, get_state_dict, freeze, unfreeze`
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training. 2020-11-29 16:12:41 -08:00			`from .model_ema import ModelEma, ModelEmaV2`
Remove commented code, add more consistent seed fn 2021-04-12 09:51:36 -07:00			`from .random import random_seed`
Reorg of utils into separate modules 2020-09-07 13:56:00 -07:00			`from .summary import update_summary, get_outdir`