You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am trying to compile and run my custom model. I use custom-model-onnx.ipynb notebook as a basis.
It compiles fine to onnx format. After several modifications, all layers are now supported by TIDL. Nevertheless, when I do
where y is a torch.Tensor of size (1, 3, 24, 9120), grid is a torch.Tensor of shape (1, 3, 2, 9120). Therefore, here two tensors of the exact same shape (1, 3, 4, 9120) are added together.
Hello.
I am trying to compile and run my custom model. I use
custom-model-onnx.ipynb
notebook as a basis.It compiles fine to onnx format. After several modifications, all layers are now supported by TIDL. Nevertheless, when I do
in the error log I have
By commenting in and out some parts of my code I figured out that something illegal happens in this code snippet:
where
y
is atorch.Tensor
of size(1, 3, 24, 9120)
,grid
is atorch.Tensor
of shape(1, 3, 2, 9120)
. Therefore, here two tensors of the exact same shape(1, 3, 4, 9120)
are added together.However, when I skip addition and leave only
it compiles and runs without errors.
My compilation parameters are the following
Have no idea how to fix it myself. Is there any way to debug it?
Thanks.
The text was updated successfully, but these errors were encountered: