From 08256aed95f4068c50fb1d5cc9239dff2a6903e8 Mon Sep 17 00:00:00 2001 From: shaikh58 Date: Thu, 19 Dec 2024 15:53:08 +0400 Subject: [PATCH] clean up print statement (#104) --- dreem/inference/post_processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dreem/inference/post_processing.py b/dreem/inference/post_processing.py index d87e8a0..cae071c 100644 --- a/dreem/inference/post_processing.py +++ b/dreem/inference/post_processing.py @@ -159,7 +159,7 @@ def filter_max_center_dist( valid = dist.squeeze() < max_center_dist # n_k x n_nonk valid_mult = valid.float().unsqueeze(-1) if valid.ndim == 1 else valid.float() - print(dist.shape, valid_mult.shape, id_inds.shape) + # print(dist.shape, valid_mult.shape, id_inds.shape) valid_assn = ( torch.mm(valid_mult, id_inds.to(valid.device)).clamp_(max=1.0).long().bool() ) # n_k x M