!Ei) SSKrSSKJr SSKJr S/rSSjr\"SS5r\"SS 5r\"S S 5r \"S S 5r Sr S\ \ S\ \ S\ \ 4SjrS\\\4S\SS4Sjrg)N)repeat)Any'consume_prefix_in_state_dict_if_presentc ^U4SjnXlU$)Nc>[U[RR5(a [ U5$[ [ UT55$N) isinstance collectionsabcIterabletupler)xns S/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/nn/modules/utils.pyparse_ntuple..parse s4 a11 2 28OVAq\"")__name__)rnamers` r_ntupler s# N Lr_single_pair_triple _quadruplec@^[U4Sj[U555$)zReverse the order of `t` and repeat each element for `n` times. This can be used to translate padding arg used by Conv and Pooling modules to the ones used by `F.pad`. c3L># UHn[T5Ho!v M M g7fr)range).0r_rs r (_reverse_repeat_tuple.. s:KqqAKs!$)r reversed)trs `r_reverse_repeat_tupler(s :HQK: ::rout_sizedefaultsreturnc"SSKn[U[UR45(aU$[ U5[ U5::a[ S[ U5S-35e[ X[ U5*SSS9VVs/sH up4UbUOUPM snn$s snnf)Nrz#Input dimension should be at least rF)strict)torchr intSymIntlen ValueErrorzip)r)r*r.vds r_list_with_defaultr6#s(S%,,/00 8}H %>s8}q?P>QRSSCM>+;" [ US5(a[UR R55nUHwn[U5S:XaMX1RSS5:XdURU5(dMAU[U5SnUR R U5UR U'My gg)axStrip the prefix in state_dict in place, if any. .. note:: Given a `state_dict` from a DP/DDP model, a local model can load it by applying `consume_prefix_in_state_dict_if_present(state_dict, "module.")` before calling :meth:`torch.nn.Module.load_state_dict`. Args: state_dict (OrderedDict): a state-dict to be loaded to the model. prefix (str): prefix. N _metadatar.)listkeys startswithr1pophasattrr:replace)r7r8r>keynewkeys rrr0s  ! "D >>& ! !V'F!+!4J  z;''J((--/0C 3x1}nnS"--1G1GS[]+/9/C/C/G/G/L $$V,(r)r)r itertoolsrtypingr__all__rrrrrr(r=r/r6dictstrrrrrKs 5 5 !Y 7 !Y  Q % ; c d3i DI "MS#X"M "M "Mr