]iSrSSKrSSKrSSKJr "SS\R 5r"SS\R 5r"SS \R 5r"S S \R 5r "S S \R 5r g)zData augmentations. N)nnc6^\rSrSrSrSU4SjjrSrSrU=r$)Shiftz8 Randomly shift audio in time by up to `shift` samples. c:>[TU]5 XlX lgN)super__init__shiftsame)selfr r __class__s K/mnt/rpi/tmp/demucs-venv-sys/lib/python3.13/site-packages/demucs/augment.pyr Shift.__init__s   cUR5up#pEXPR- nURS:aUR(d USSU24nU$UR(aSOUn[R "URX'SS/UR S9nURSX4S5n[R"XaR S9n URSX-5nU$)Nr.device) sizer trainingr thrandintrexpandarangegather) r wavbatchsourceschannelstimelengthsrcsoffsetsindexess rforward Shift.forwards),& " ::>==#ww,' !IIq7**TZZ%q!1DSZZX!..WC))F::>jjG$56 r)r r )i F __name__ __module__ __qualname____firstlineno____doc__r r(__static_attributes__ __classcell__rs@rrrs  rrc\rSrSrSrSrSrg) FlipChannels&z Flip left-right channels. c^UR5up#pEUR(aURS5S:Xas[R"SX#SS4URS9nUR SSSU5nSU- n[R "URSU5URSU5/SS9nU$)Nrrrdim)rrrrrrcatr)r rr r!r"r#leftrights rr(FlipChannels.forward*s),& ==SXXa[A-::a%!Q!7 KD;;r2r40DHE&&#**Q-szz!U/CD!LC rNr+r,r-r.r/r(r0r>rrr4r4&s rr4c\rSrSrSrSrSrg)FlipSign4z Random sign flip. cUR5up#pEUR(a=[R"SX#SS4UR[R S9nUSU-S- -nU$)Nr7r)rdtype)rrrrrfloat32)r rr r!r"r#signss rr(FlipSign.forward8sV),& ==JJq51a"8SUS]S]^EUQ'C rr>Nr?r>rrrArA4s rrAc6^\rSrSrSrSU4SjjrSrSrU=r$)Remix@z$ Shuffle sources to make new mixes. c:>[TU]5 XlX lg)a| Shuffle sources within one batch. Each batch is divided into groups of size `group_size` and shuffling is done within each group separatly. This allow to keep the same probability distribution no matter the number of GPUs. Without this grouping, using more GPUs would lead to a higher probability of keeping two sources from the same track together which can impact performance. N)r r proba group_size)r rLrMrs rr Remix.__init__Ds  $rc UR5up#pEURnUR(a[R"5UR:aUR =(d UnX'-S:wa[ SUSU35eX'-nURXX4U5n[R"[R"XUSSUS9SS9n URSU RSSSXE55nURX#XE5nU$)Nrz Batch size z! must be divisible by group size rrr8r) rrrrandomrLrM ValueErrorviewrargsortrandrr) r rr streamsr"r#rrMgroups permutationss rr( Remix.forwardQs),& ==V]]_tzz91EJ!Q& ;ug5VWaVb!cdd(F((6w$GC::bggf'1aX^&_*+-L**Q 3 3BB OPC((58:C r)rMrL)rr*r2s@rrIrI@s %rrIc2^\rSrSrSU4SjjrSrSrU=r$)ScalebcF>[TU]5 XlX lX0lgr)r r rLminmax)r rLr^r_rs rr Scale.__init__cs  rc*UR5up#pEURnUR(ab[R"5UR:a?[ R "X#SSUS9RURUR5nX-nU$)Nrr) rrrrPrLremptyuniform_r^r_)r rr rUr"r#rscaless rr( Scale.forwardisk),& ==V]]_tzz9XXea6BKKDHHVZV^V^_F MC r)r_r^rL)g?g?g?)r+r,r-r.r r(r0r1r2s@rr[r[bs rr[) r/rPtorchrrModulerr4rArIr[r>rrrhsf BII0 299  ryy BIID BII r