FBallert comments

Repositories
Issues
Comments

Results 1 comments of


                                            FBallert

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())` Line 243: `cov_inverse =...