Traceback (most recent call last):
File "test.py", line 4, in
plot_example(True, reference, translation)
File "/home/jesse/.local/lib/python3.7/site-packages/moverscore_v2.py", line 194, in plot_example
distance_matrix = masks * distance_matrix
RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Long
(The line in question is
distance_matrix = masks * distance_matrix,
when i change it to distance_matrix = masks.double() * distance_matrix.double(),
I get this error:
Traceback (most recent call last): File "test.py", line 4, in
plot_example(True, reference, translation)
File "/home/jesse/.local/lib/python3.7/site-packages/moverscore_v2.py", line 194, in plot_example
distance_matrix = masks * distance_matrix
RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Long
(The line in question is distance_matrix = masks * distance_matrix,
when i change it to distance_matrix = masks.double() * distance_matrix.double(),
the problem seems to disappear)