search

yoyololicon / pytorch-NMF

A pytorch package for non-negative matrix factorization.
https://pytorch-nmf.readthedocs.io/
MIT License
226 stars 24 forks source link

Fail to import #18

Closed RuiFeiHe closed 3 years ago

RuiFeiHe commented 3 years ago

from torchnmf.nmf import NMF *** torch.jit.frontend.NotSupportedError: unsupported kind of augumented assignment: BitOr: File "/home......../lib/python3.7/site-packages/torchnmf/nmf.py", line 44 break

    zero_coef |= mask
              ~~ <--- HERE
    v.relu_()
    v += (k1 - v.sum()) / (N - zero_coef.count_nonzero())
RuiFeiHe commented 3 years ago

solved. |= -> = ... |

yoyololicon commented 3 years ago

@RuiFeiHe Update pytorch to latest version should solve the issue.