hidet-org / hidet

An open-source efficient deep learning framework/compiler, written in python.

Home Page:https://hidet.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[FEATURE] Meet an undefined operator when compiling NASNet

VincentXWD opened this issue · comments

hello, I'm using hidet to compile PNASNET-5-LARGE defined here https://github.com/Cadene/pretrained-models.pytorch#pnasnet

And I met an undefined operator 'torch.nn.ZeroPad2d':

NotImplementedError: The following modules/functions are not supported by hidet yet:
torch.nn.ZeroPad2d

Could someone please add this module into hidet? thanks.

Hi @VincentXWD,

I have added the missing operators in #399, could you check whether it works?

Thank you @yaoyaoding, I'll check it later.

Happy holidays.

@yaoyaoding Thanks, it works. Good job!