Yue

icon indicating a website link https://yuehu9.github.io/

icon company Vanderbilt University

Results 1 comments of Yue

DGAMM-LSTM cannot run on gpu

> I faced the same issue with the plain DAGMM. Worked around this by adding .cpu() and .cuda() as follows in src/algorithms/dagmm.py: > > Line 231: `pinv = np.linalg.pinv(cov_k.data.cpu().numpy())` >...