From f8a587de510902a88d2b09d3aab4e894e7990b7f Mon Sep 17 00:00:00 2001 From: MaksimEkin Date: Wed, 10 Jan 2024 13:59:09 -0700 Subject: [PATCH] hot fix wnmf --- TELF/factorization/NMFk.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TELF/factorization/NMFk.py b/TELF/factorization/NMFk.py index 73a59b79..d325234c 100644 --- a/TELF/factorization/NMFk.py +++ b/TELF/factorization/NMFk.py @@ -152,6 +152,8 @@ def _perturb_parallel_wrapper( if use_gpu: Y = __put_X_gpu(Y, gpuid) W_init, H_init = __put_WH_gpu(W_init, H_init, gpuid) + if "WEIGHTS" in nmf_params and nmf_params["WEIGHTS"] is not None: + nmf_params["WEIGHTS"] = __put_X_gpu(nmf_params["WEIGHTS"], gpuid) W, H, other_results = __run_nmf(Y, W_init, H_init, nmf, nmf_params, use_gpu, gpuid)