o ÓÙ¾iâã@s"dZddlZd dd„Zdd„ZdS) zC Batch size decay and retry helpers. Copyright 2022 Ross Wightman éNéFcCsj|dkrdStdt |d¡t d¡ƒ}t||dƒ}|||d||}|r3|dr3|d8}|S)ay power of two batch-size decay with intra steps Decay by stepping between powers of 2: * determine power-of-2 floor of current batch size (base batch size) * divide above value by num_intra_steps to determine step size * floor batch_size to nearest multiple of step_size (from base batch size) Examples: num_steps == 4 --> 64, 56, 48, 40, 32, 28, 24, 20, 16, 14, 12, 10, 8, 7, 6, 5, 4, 3, 2, 1 num_steps (no_odd=True) == 4 --> 64, 56, 48, 40, 32, 28, 24, 20, 16, 14, 12, 10, 8, 6, 4, 2 num_steps == 2 --> 64, 48, 32, 24, 16, 12, 8, 6, 4, 3, 2, 1 num_steps == 1 --> 64, 32, 16, 8, 4, 2, 1 érr)ÚintÚmathÚlogÚmax)Ú batch_sizeÚnum_intra_stepsÚno_oddÚbase_batch_sizeÚ step_size©r úJ/home/ubuntu/.local/lib/python3.10/site-packages/timm/utils/decay_batch.pyÚdecay_batch_steps   rcCs$| ¡}d|vr dSd|vrdSdS)z_ check failure error string for conditions where batch decay retry should not be attempted z required rankFÚillegalT)Úlower)Ú error_strr r rÚcheck_batch_size_retry s r)rF)Ú__doc__rrrr r r rÚs