o si@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZmZgd ZddZddZdS)) BaseModel) ConvTasNet)DCCRNet)DCUNet) DPRNNTasNet)SuDORMRFImprovedNet SuDORMRFNet)DPTNet) FasNetTAC) LSTMTasNet)DeMask)XUMX)save_publishableupload_publishable) rrrrr r r r rrr rrcCsL|jtvs|jtvrtd|jdt|j|idS)zuRegister a custom model, gettable with `models.get`. Args: custom_model: Custom model to register. zModel z% already exists. Choose another name.N)__name__globalskeyslower ValueErrorupdate) custom_modelrL/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/models/__init__.pyregister_model"srcCs\t|tr%ddtD}||}|dur#tdt||Stdt|)zReturns an model class from a string (case-insensitive). Args: identifier (str): the model name. Returns: :class:`torch.nn.Module` cSsi|] \}}||qSr)r).0kvrrr ;szget..Nz!Could not interpret model name : ) isinstancestrritemsgetrr) identifierto_getclsrrrr!1s r!N) base_modelsr conv_tasnetrdccrnetrdcunetr dprnn_tasnetrsudormrfrrdptnetr fasnetr lstm_tasnetr demaskr x_umxr publisherrr__all__rr!rrrrs