]iASrSSKrSSKJr SSKJr SSKJrJr SSK r SSK J s J r SSKJrJrJrJr SSKJr SS KJr SS KJrJr SS KJr SS KJrJr \R<"\5r S r!"SS\"5r#g)zMain training loop.N)get_xp)write_and_rename) LogProgressbold)augmentdistribstates pretrained) apply_model)ModelEMA)evaluatenew_sdr) svd_penalty) pull_metricEMAcNSRSUR555$)N | c3P# UHupUR5SU3v M g7f)=N) capitalize).0keyvals J/mnt/rpi/tmp/demucs-venv-sys/lib/python3.13/site-packages/demucs/solver.py _summary..s%R/hc)*!C51/s$&)joinitems)metricss r_summaryr!s ::R'--/R RRcZ\rSrSrSrSrSrS\S\4SjrS\S\4Sjr S r S S jr S r g )Solverc TX@lXlX lX0l[R "URUR UR5Ul[R"U5Ul [[URR555RUl//S.UlURR!5Hon[#UR$U5nUS:Xa UROSnU(dM6UH3nURUR'[)URXS95 M5 Mq [*R,"[/UR0R2UR0R4-5UR*R6S9/n UR*R8(a-U [*R:"5[*R<"5/- n SH_n [#UR*U 5n U R>(dM,U R'[#[*U RA55"S 0U D65 Ma [BRDRF"U 6Ul[I5n U RJUl%U RJS- Ul&U RJS- Ul'[PRSS URLRU55 SUl+S Ul,U RZUl-URZR\Ul.UR_5 g) N)batchepochr'cpu)device)shiftsame)scaleremix checkpoint.thzbest.thzCheckpoint will be saved to %sF)0argsloadersmodel optimizerr get_quantizerquant quantizerr wrapdmodelnextiter parametersr*emaskeysgetattremaappendr rShiftintdset samplerater+ shift_sameflip FlipChannelsFlipSignprobartorchnn Sequentialrfoldercheckpoint_file best_fileloggerdebugresolve best_state best_changedlinkhistory_reset) selfr2r3r4r1kinddecaysr*decayaugmentsaugkwxps r__init__Solver.__init__s2   "--djj$**dnnUll5) 4 5 5 789@@ !2. IINN$DTXXt,F$(GOT[[Fv#EIIdO**8DJJ+UV$ %MMDII,@,@499??,R(S'+||'>'>@A <<   --/1A1A1CD DH%Cs+Bxxx1A B HR HI&xx**H5 Xii !yy?:Y. 5t7K7K7S7S7UV!GG yy((  r"c\0nURR5US'URR5US'URUS'URUS'UR US'UR R5H2up4[U5HupVUR5USUSU3'M M4 [UR5n[R"X'5 SSS5 UR RnU(agUS-U-S :Xa[US-UR R:wa>[URS US-S 3- 5n[R"X'5 SSS5 UR (av[UR"5n[$R&"URUR 5nURUS'[R"X'5 SSS5 S Ulgg!,(df  GN=f!,(df  N=f!,(df  N:=f) Nstater4rWrTr1ema__rr checkpoint_z.thF)r3 state_dictr4rWrTr1r=r enumeraterrOrKsave save_everyepochsrNrUrPr serialize_model) rYr(packagerZr=kr@tmprks r _serializeSolver._serializeMs::002#~~88: !\\  $ ))))//+JD#D/-0^^-=$tfAaS)**,d22 3s JJw $4YY)) 519 2a7EAIIYIY> %!'$-7F4L W #GFOU;F6N W #GFOU;F6N g %g.F7O   &wy'95 AF9  g $WW%5u=F7O r"c0nSU;a[USS5US'SU;a[USS5US'URRH9nSU3nXA;a[XS5X$'SU3nXA;dM)[XS5X$'M; U$)zFormatting for test metrics.sdrrrsdr_nsdr_)rr3sources)rYr rsourcers r _format_testSolver._format_tests G "75>59F5M W #GFOU;F6Njj((F/C~$W\59 &"C~$W\59 ) r"c TUR(a[RS5 [UR5HupUR US5n[R[ SUS-S[ U5355 UR US5n[R[ SUS-S[ U5355 SU;dMURUS5nU(dM[R[ S US-S[ U5355 M S n[[UR5URR5GHnURR5 0n[RS 5 [RS 5 URU5US'UR US5n[R[ SUS-S[ U5355 [RS 5 [RS 5 URR5 [ R""5 URUSS9nUnSn[$R&"URR)55n0US'XBSS'URR*R,nUR.R15Hup[U 5HxupU R35 URUSS9nSSS5 SU SU 3n XBSU 'XHnXXnUR5S5(aU*nU*nX:dMhUnU R6nU nMz USR9U5 XbSS'M SSS5 USWn[;UR<RSU35U/-nUR5S5(a [?U5nO [AU5nUUSS'URRBRDS :an[GURRB5nURIS5 [ R""5 [KUR4SS0UD6nSSS5 WUSS'UR US5n[R[ SUS-S[ U5355 UU:Xd%URRLRN(aA[R[ S5U5 [$R&"W5Ul(SUl)US-URR*RT-S :HnXRRS- :HnU(dU(GaL[RS 5 [RS5 URR*RV(a URPnO.[$R&"URR)55nURR*RX=(a Un[$RZ"URU5 [ R""5 []UUS9US'SSS5 SSS5 URUS5n[R[ S US-S[ U5355 UR<R_U5 [`RbS :Xa?UReU5 [RgSURhRk55 U(dGM g g!,(df  GNF=f!,(df  GN=f!,(df  GN=f!,(df  GN=f!,(df  GN=f)Nz#Replaying metrics from previous runtrainzTrain Summary | Epoch rrvalidzValid Summary | Epoch testzTest Summary | Epoch rzF----------------------------------------------------------------------z Training...zCross validation...F)rmainrerfrrzvalid.rrexactTzNew best valid loss %.4fzEvaluating on the test set...) compute_sdrzCheckpoint saved to %s)6rWrQrxrirrr!rrangelenr1rlr3r_run_one_epochevalrKno_gradr copy_staterhrmetricr=rswap startswithrdupdaterrVmaxminsvdrdictpoprrD train_validrTrUeveryrr swap_stater push_metricsr rankrqrRrOrS)rYr(r formattedrbvalidrrdrrZr=ror@rtab valid_lossmets best_lossr_r should_evalis_lastrs rr Solver.trains << KK= >' 5NE**77+; 9KK'.>?E JJ   G KK ! KK &#2259GG **77+;>&11!"A!"A5%*F$'IIE$(E#2G$++F305G$W-!#4!4!)#.Jtyy00F3%.AZLPD~~f%%I I '0GG V $yy}}$$q($))--(y!]]_)$**GDGBGG%.5 +**77+;>&& OOE"--djj.C.C.EFE"iinn00  D#8 3xPYGZF[!\]^ II " "7 +||q & 5t7K7K7S7S7UVwI@2(Z!H%_L):9s[B.]"] >]"3]"]4^1^?^ ]]"" ]1 4 ^  ^ ^ ^' cURnU(aURSOURSn[RS:a"U(aURR U5 SS/UnUSUS-3-n[ U5nUR(a[XsR5n[[XGURRRUS9n[5n [U5GHupU RUR 5n U(a!UR#U 5n U R%SS9n OU SS2S 4n U SS2SS24n U(dOURR&(a4[)UR*XRR,R.S S 9n OUR1U 5n U(a6[3UR*S 5(aUR*R5X5n U R6U R6:XdU R6U R645e[9[;S U R=555nUR>R@S :Xa8[BRD"XSS9nURGU5RGS 5nUnOyUR>R@S:Xa=[BRH"XSS9nURGU5nUS-nURGS 5nO"[KSURR@35e[LRN"URP5RU 5nUU-R%5UR%5- nS nURRbURRRU5nURVRX(aXRVRXU-- n0nUU-R%5UR%5- US'UUS'U(d|[[XR]55RGS 5nS n[_UR*R`UU5HunnnUUSU3'UUU-- nM UUR%5- US'U(avURbRdS :a\[gURb5nURiS5 [kUR*40UD6nUUS'XRbRdU-- nUUS'[UR*R`5HunnUUUSU3'M U(GaURm5 S n/nUR*Ro5HanURpcMUURpRrRu5S -- nURwURpRr5 Mc US-US'UR>Rx(aV[LRzR|RUR*Ro5UR>Rx5 URRS:Xa?UR*R5H!unnURpbM[SU5 M# URR5 URR5 URSHnUR5 M U "U5nURU5n UR"S 0U D6 AA AAURU :Xa OEURR(a U(a O!URRS:XdGM O U(a&URSHnUR5 M [R"WW S-5$)!NrrrValidTrainz | Epoch )totalupdatesrt)dimr)splitoverlaptransform_targetl1none) reductionmseg?z Invalid loss rrrrrrreco_runszno gradr'rRr(r0)Kr1r2r world_sizesampler set_epochr max_batchesrrrQmisc num_printsrritor*rsum valid_applyr r3rrr9hasattrrshapetuplerroptimrFl1_lossmeanmse_loss ValueErrorrKtensorweightsr7 model_sizer6diffqrdetachziprrrrrrbackwardr<rdatanormrA clip_gradrLutilsclip_grad_norm_flagnamed_parametersprintr4step zero_gradr=rrrRaverage)!rYr(rr1 data_loaderlabelrtrlogprogaverageridxrmixestimatedimsrrrrrnsdrsrrwr_rro grad_normgradspnr@logss! rrSolver._run_one_epoch#syy/4dll7+$,,w:O    !e    ) )% 0'"5)519+..K    //0Efk&*iinn&?&?dL5%g.LCjj-G,,w/kkak(adm!!QR%.TYY22&tzz3iinn>R>R\]^;;s+-?@@**55cC>>W]]2 SX^^W]]4S S2q'++-01Dzz$&yyfEyy++A.E)zz(vFyySyyy| =0@!ABBll4<<033G>U* $ ; ; =166>!)Q/!>##%((*99W-CJJL.f%F%%f-D NN "T "hb3&yy5yy~~(/@ yy) *vsQw//r")r1rrUrPrTrOr*r9r=rNrWrVr2r3r4r7N)T) __name__ __module__ __qualname____firstlineno__rarqrXrrrrr__static_attributes__r0r"rr$r$sF,\&4E>Td,DT unr0r"r$)$__doc__loggingdorar dora.utilsrdora.logrrrKtorch.nn.functionalrL functionalrrr r r applyr r@r rrrrrrr getLoggerrrQr!objectr$r0r"rrsX '& 22'#   8 $Sw0Vw0r"