Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GoogleNet aux is None in pytorch test #6201

Closed
wjliu1998 opened this issue Aug 4, 2020 · 2 comments
Closed

GoogleNet aux is None in pytorch test #6201

wjliu1998 opened this issue Aug 4, 2020 · 2 comments

Comments

@wjliu1998
Copy link
Contributor

When running
TVM_FFI=ctypes python -m pytest -v /home/liuwenjian/incubator-tvm/tests/python/frontend/pytorch/test_forward.py -k "test_googlenet"

error will occur at
https://github.com/apache/incubator-tvm/blob/90bde33e60a3ae0d18a767eba9a2092eab204788/tests/python/frontend/pytorch/test_forward.py#L179

the error message is
"
assert(isinstance(orig, torch.nn.Module))
E AssertionError
anaconda3/lib/python3.7/site-packages/torch/jit/init.py:2195: AssertionError
"

This is because when getting googlenet pretrained model, the aux module is always None. Therefore the assert will fail

torch.version='1.6.0+cu101'
tvm.version='0.7.dev1'

@masahi
Copy link
Member

masahi commented Aug 4, 2020

PyTorch 1.6 is not supported yet, a PR welcome.

@wjliu1998
Copy link
Contributor Author

I'll try this one, thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants