siju-samuel commented on issue #5133:
URL: https://github.com/apache/incubator-tvm/issues/5133#issuecomment-617591097
@gopinath-r
Currently tvm-pytorch frontend doesnt support taking each value of `reps` as
another tensor or function. It is expecting as an simple constant and in this
case we are getting a multiplication operator.
So as a workaround, you can modify your `bts.py`
change your `local_planar_guidance` method
typecast the scalar multiplication to int.
Modify the below two lines and do a simple typecast before multiplication.
u = self.u.repeat(plane_eq.size(0), plane_eq.size(2) *
int(self.upratio), plane_eq.size(3))
v = self.v.repeat(plane_eq.size(0), plane_eq.size(2),
plane_eq.size(3) * int(self.upratio))
to
u = self.u.repeat(plane_eq.size(0), int(plane_eq.size(2) *
int(self.upratio)), plane_eq.size(3))
v = self.v.repeat(plane_eq.size(0), plane_eq.size(2),
int(plane_eq.size(3) * int(self.upratio)))
Hope this will resolve your issue.
Please use https://discuss.tvm.ai/ from now onwards for discussing bugs and
issues.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
