The traceback indicates that (in sparse.py) you’re trying to index_select weight with something that’s out of range. What is the shape of your input to forward?
1 Like
The traceback indicates that (in sparse.py) you’re trying to index_select weight with something that’s out of range. What is the shape of your input to forward?