From b51ce277827ecb93afb9c0f81bfd00800bcf334f Mon Sep 17 00:00:00 2001 From: hyunp2 <42776897+hyunp2@users.noreply.github.com> Date: Sat, 2 Sep 2023 22:27:07 -0500 Subject: [PATCH] Update datasets.py --- utils/src/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/src/datasets.py b/utils/src/datasets.py index 709fc430..4ab9a0bf 100644 --- a/utils/src/datasets.py +++ b/utils/src/datasets.py @@ -346,7 +346,7 @@ def collate(batch): out[key] = torch.nn.utils.rnn.pad_sequence(value, batch_first=True, padding_value=0) continue # raise Exception(f'Unknown batch key: {key}') - print(f"Passing {key}") + # print(f"Passing {key}") atom_mask = (out['fragment_mask'].bool() | out['linker_mask'].bool()).to(const.TORCH_INT) out['atom_mask'] = atom_mask[:, :, None]