hS %SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJ r SSK J r SSK J r JrJrJrJrJrJrJrJrJrJr SSKrSSKJr SSKJrJrJr SSKJ r SS K!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+ SS K!J,r- SS K!J.r/ SS K!J0r0 \ "S SS9r1\ \2\3S 'SSK J4r4 SSK5J6r6 SSK6J7r7J8r8J9r9J:r:J;r;Jr> ShSjr?Sr@SrASiS\BSS4SjjrCS\ S\24SjrDS\ S\24SjrES\ S\24SjrFS\ S\24SjrGS\ S\24S jrHS\ S\24S!jrIS\ S\24S"jrJS\ S\24S#jrKS\ S\24S$jrLS\ S\24S%jrMS\ S\24S&jrNS'rOS(\BSS)4S*jrPS\24S+jrQSiS(\BS\24S,jjrRSiS(\BS\24S-jjrSS.\8S/\8SS4S0jrTSjS1S2.S3\:S4\\BS5\US\94S6jjjrVS7S8.S9\\8\<\;\\84S:\BS\B4S;jjrWSkS<jrXSkS=jrYSkS>jrZS?\\ /\24S@\\ /\ 4S\ S\ 4SAjr[S?\\ /\24S\ S\ 4SBjr\SlSC\8SD\2SE\SSSF4SGjjr]SSH.SISFSJ\SKS\84SLjjr^SmSC\8SD\2SM\2SSN4SOjjr_SSH.SPSNSJ\SKS\84SQjjr`SnSC\8SD\2SSR4SSjjraSSH.STSRSJ\SKS\84SUjjrb\"SV5rcSW\SX\c4SY\ SZ\ S\SX\c44S[jrd"S\S]\e5rfS^\gSW\\ \ \2/\ 4S9\ S3\ SS4 S_jrh\RSoS`j5rj\RSa5rk\RSpSb\gSc\B4Sdjj5rl\ "SeSf55rm/SgQrng)qN) ContextVar) dataclass) AnyCallableDictListMappingOptionalSequenceTupleTypeVarUnioncast)Version) ConfigDictValidationError create_model)table) cupycupy_from_dlpackhas_cupy has_cupy_gpuhas_gpu has_mxnethas_tensorflow has_torchhas_torch_cuda_gpu has_torch_mps)mxnet) tensorflow)torchDATA_VALIDATIONF)default) TYPE_CHECKING)types) ArgsKwargsArrayXdFloatsXdIntsXdPaddedRagged)Opsreturn torch.devicec^[c [S5eSSKJn SSKJn SSKJn U"5n[X15(a7[RR5n[R"SU35$[X25(a[R"S5$[R"S5$) Nz[U5=(a UR$rK)rais_cudarYs r@is_torch_cuda_arrayrds #  .3;;.rBc<[U5=(d [U5$rK)rdis_torch_mps_arrayrYs r@is_torch_gpu_arrayrgs s # >'9#'>>rBcd[U5=(a [US5=(a UR$)Nis_mps)rahasattrrirYs r@rfrfs" #  H73#9 HcjjHrBc\[(dg[U[R5(aggr_)rr:tfr`rYs r@is_tensorflow_arrayrms! > C # #rBcD[U5=(a SUR;$)NzGPU:)rmr=rYs r@is_tensorflow_gpu_arrayros s # <#**(<s r@ require_cpurs2 %.CC rBc6[(a [US9 [$)z?Use GPU if it's available. Returns True if so, False otherwise.r~)r require_gpurs r@ prefer_gpursw6" NrBcSSKJnJnJn [R "5S:Xa,[ (d![(a [S5e[S5e[R "5S:wa[(d [S5e[(d [S5e[(aU"U"55 [U5 gU"U"55 g) Nr)r2r3rDarwinz6Cannot use GPU, installed PyTorch does not support MPSz(Cannot use GPU, PyTorch is not installedz%Cannot use GPU, CuPy is not installedzNo GPU devices detectedT) r7r2r3rplatformsystemrrr6rrrr)r~r2r3rs r@rrs::H$]] 9UV VCDD  h &xx@AA W233| "v  ! rBdstsrcc.[U[R5(a$[U[R5(aXSS&g[U5(a,[R "USS9n[R "X5 g[R "X5 g)NF)copy)r:rEr\rFrr{copyto)rrs r@ copy_arrayrs]#u}}%%*S%--*H*HA s  jj5) C SrB)label_smoothingY n_classesrc Uc"[[R"U5S-5nUS:a [S5eUS:XaUS:Xa [S5eSnOUS:d[SUS35eX!S- - nUS- U- nUS:aX$:a[SUS US US 35e[ U5nUR X4US S 9nUR USU- 5 X`$)Nrrz>Label-smoothing parameter has to be greater than or equal to 0rzn_classes should be at least 1zGn_classes should be greater than 1 when label smoothing is enabled,but z was provided.zFor z7 classes label_smoothing parameter has to be less than z , but found .float32)dtype)intrEmaxr6rIfull fill_diagonal)rrr nongold_prob max_smoothxp label_distrs r@to_categoricalrs  ! q()  L  # >=> > 1} k1 'a-8 a-9,J1}69+=l,&7q :  ! B''90,i'PK[!o"56 >rBdimXrcP[U[5(a[URUS9$[U[5(a[URUS9$[ US5(a[ US5(an[ [U5n[UR5S:Xag[UR5S:Xa[UR55S-$URU$[U[[45(a[U5S:Xag[USUS9$Sn[U5e)zjInfer the 'width' of a batch of data, which could be any of: Array, Ragged, Padded or Sequence of Arrays. rshapendimrrz=Cannot get width of object: has neither shape nor __getitem__)r:r, get_widthr|r+rjrr(lenrrrlisttupler6)rrerrs r@rr s !VS)) Av  S)) G  F!3!3 !  qww<1  \Q quuw=2.0.0,<2.6.0)pkgN)r ImportErrorformat)templates r@assert_tensorflow_installedr<s+RH >(//.H/IJJ rBc0[(d [S5eg)z/Raise an ImportError if MXNet is not installed.zjMXNet support requires mxnet: pip install thinc[mxnet] Enable MXNet support with thinc.api.enable_mxnet()N)rrrLrBr@assert_mxnet_installedrCs 9 z   rBc0[(d [S5eg)z1Raise an ImportError if PyTorch is not installed.z8PyTorch support requires torch: pip install thinc[torch]N)rrrLrBr@assert_pytorch_installedrKs 9TUU rBis_match convert_itemc8^^T"U5(aT"U5$[U[5(a:[TT[UR 555n[R "U5$[U[ 5(a;0nUR 5H#upE[TTU5n[TTU5nXSU'M% U$[U[5(aUVs/sHn[TTU5PM sn$[U[5(a[UU4SjU55$U$s snf)zEither convert a single value if it matches a given function, or recursively walk over potentially nested lists, tuples and dicts applying the conversion, and returns the same type. Also supports the ArgsKwargs dataclass. c3>># UHn[TTU5v M g7frK)convert_recursive).0itemrrs r@ $convert_recursive..hs UQT&xtDDQTs)r:r'rritems from_itemsdictr)rrrW convertedkeyvaluers`` r@rrQs}}C  C $ $%h d399;>OP $$Y// C   ))+JC#HlC@C%h eDE"cN& C  LOPCD!(L$?CPP C  UQTUUU QsDc## U"U5(aUv g[U[5(a,[U[UR 555ShvN g[U[ 5(a@UR 5H+up#[X5ShvN [X5ShvN M- g[U[5(d[U[ 5(aUHn[X5ShvN M ggNNfNUN7f)zEither yield a single value if it matches a given function, or recursively walk over potentially nested lists, tuples and dicts yielding matching values. Also supports the ArgsKwargs dataclass. N)r:r'iterate_recursiverrrr)rrWrrrs r@rrms }} C $ $$XtCIIK/@AAA C  ))+JC(7 7 7(9 9 9& C  *S%"8"8D(8 8 8#9 B 8 9 9sIAC;C3=C;C5C;!C7"AC;(C9) C;5C;7C;9C; xp_tensor requires_gradr=z torch.Tensorc[5 Uc [5n[US5(a:UR5n[R R RU5nOQ[US5(a*[R R RU5nO[R"U5nURU5nU(aUR5 U$)z3Convert a numpy or cupy tensor to a PyTorch tensor.toDlpack __dlpack__) rrArjrr"utilsdlpack from_dlpack from_numpytorequires_grad_)rrr= dlpack_tensor torch_tensors r@xp2torchrs  ~)+y*%%!**, {{))55mD L ) ){{))55i@ '' 2 ??6*L##% rB)r>rr>r-cSSKJn [5 [U5(an[ X5(a,UR 5R 5R5$[[RRRU55$[ X5(dUc,UR 5R 5R5$[R"U5$)zConvert a torch tensor to a numpy or cupy tensor depending on the `ops` parameter. If `ops` is `None`, the type of the resultant tensor will be determined by the source tensor's device. rNumpyOps)apirrrdr:detachr5rErr"rr to_dlpackrasarray)rr>rs r@torch2xprs <(( c $ $&&(,,.446 6#EKK$6$6$@$@$NO O c $ $ &&(,,.446 6<< - -rB as_variablez tf.Tensorc[5 [US5(a:UR5n[RR R U5nOa[US5(a:UR5n[RR R U5nO[R"U5nU(a=[R"UR5 [R"XAS9nSSS5 USLaFUSLaA[R"UR5 [R"U5nSSS5 U$U$!,(df  N[=f!,(df  U$=f)zAConvert a numpy or cupy tensor to a TensorFlow Tensor or Variablerr) trainableNF) rrjrrl experimentalrrrconvert_to_tensorr=Variable stop_gradient)rrrr tf_tensors r@ xp2tensorflowrs  !y*%%!**, OO**66}E L ) )!,,. OO**66}E ((3 YYy'' ( IGI)+"6YYy'' (((3I) 9) ( ) ( sE%E E E(rcxSSKJn [5 [U5(aT[ X5(aUR 5$[ RRRU5n[U5$[ X5(dUcUR 5$[R"UR 55$)zConvert a Tensorflow tensor to numpy or cupy tensor depending on the `ops` parameter. If `ops` is `None`, the type of the resultant tensor will be determined by the source tensor's device. rr) rrrror:rErlrrrrrr)rr>rrs r@ tensorflow2xprs !y)) c $ $??$ $OO22<rs r@mxnet2xprs)$$ c $ $##%--/ /#I$A$A$CD D c $ $ ##%--/ /<< 1 1 34 4rBPartialTfunc.argskwargsc\[R"U/UQ70UD6nURUlU$)zfWrapper around functools.partial that retains docstrings and can include other workarounds if needed. ) functoolspartial__doc__)rrr partial_funcs r@rrs0 $$T;D;F;L< locmsgz  N)rGjoinrzstrappendrr6__init__) selfrrrrmessage type_infor|errorperr_locresults r@r DataValidationError.__init__s/tfA6$q'$tAwi0 Ekk599UB3G"H3Ga3q63G"HIG KK))E"23 4eDk2D&499V+<"<=#IsB> rL) __name__ __module__ __qualname____firstlineno__r rrr r rrr __static_attributes__rLrBr@rrsh LN >> >  > hwsCx014S#X3GGH >  >>rBrrcp[R"U5n[RRn[ UR R 55n[U5S:waO[U5SSRUVs/sHowRPM sn5S3nSU3n [XUSU 0/5egs snf)zValidate the input and output of a forward function against the type annotations, if available. Used in Model.initialize with the input and output samples as they pass through the network. z (z, )zJInvalid forward function. Expected 3 arguments (model, X , is_train), got rN) inspect signature Signatureemptyr parametersvaluesrr rr) rrrrsigr paramsr bad_paramsrs r@validate_fwd_input_outputr&#s   D !C    # #E #..'') *F 6{aF }Btyy&1I&Q&&&1I'J&K1M Z[eZfg!$1s|n==8 =2Js>B3 c## [R"USS9nUv UR5 [R"UR 5 g7f)NF)modedelete)tempfileNamedTemporaryFilecloseosremover)r(fs r@ make_tempfiler0Os5##e> ;; <<5W55WrBr?)rIrArVrFrDrrrrrrrrrrr&rr0r=r?rr)r.r/)rrK)r.N)FN)FF)F)r)r)o contextlibrrr-rrPr*r2 contextvarsr dataclassesrtypingrrrrr r r r r rrrEpackaging.versionrpydanticrrrwasabircompatrrrrrrrrrrr rqr!rlr"r#boolrIr%r&r'r(r)r*r+r,rr-rArIrMrrVrZrFrDrardrgrfrmrortrxr}rrrrrfloatrrrrrrrrrrrrrrrr6rr r&contextmanagerr0r7r=r?__all__rLrBr@rZs  "!     %>>    $$./@%$PD!P HH$   3# 3d 35S5T5 st/S/T/?C?D?ICIDIST===DCDDD! 3 #5 Ts4D*G'd $%! % %}% %  %RIK Wffhw&7 78BE8K V ud{#3;SE3J3GNQ89# 59C9C9( '+ ^ $ 8=A. .*25/. .,JO'+BF67;33$,UO3 3./4  '+   ;?55(05 5( :  3= !*-9< c8m>*>&) )sC.34)9<)AD) )X  "" C3  , rB