[FEATURE] Meet an undefined operator when compiling NASNet
VincentXWD opened this issue · 3 comments
VincentXWD commented
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.
yaoyaoding commented
Hi @VincentXWD,
I have added the missing operators in #399, could you check whether it works?
VincentXWD commented
Thank you @yaoyaoding, I'll check it later.
Happy holidays.
VincentXWD commented
@yaoyaoding Thanks, it works. Good job!