Haven’t tried GPU yet. I thought CPU is enough for my relatively small problem.
Besides sparse=True, I also pulled out the tensor creation from the tight loop.
However, it’s still not significantly faster. As of now, I see 2 options, I still can pursue:
A) using batches instead of feeding trigrams one by one -> does somebody know how to do this?
B) using GPU