From 87dd325b8d54a8ff8da25e7f70123b425fc5c97d Mon Sep 17 00:00:00 2001 From: Tony-HYX <605698554@qq.com> Date: Sun, 10 Dec 2023 11:17:45 +0800 Subject: [PATCH] [ENH] add filter pseudo and remove loss in log --- abl/bridge/simple_bridge.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/abl/bridge/simple_bridge.py b/abl/bridge/simple_bridge.py index 508c106..b31e284 100644 --- a/abl/bridge/simple_bridge.py +++ b/abl/bridge/simple_bridge.py @@ -76,13 +76,12 @@ class SimpleBridge(BaseBridge): self.predict(sub_data_samples) self.idx_to_pseudo_label(sub_data_samples) self.abduce_pseudo_label(sub_data_samples) + self.filter_pseudo_label(sub_data_samples) self.pseudo_label_to_idx(sub_data_samples) - loss = self.model.train(sub_data_samples) + self.model.train(sub_data_samples) print_log( - f"loop(train) [{loop + 1}/{loops}] segment(train) \ - [{(seg_idx + 1)}/{(len(data_samples) - 1) // segment_size + 1}] \ - model loss is {loss:.5f}", + f"loop(train) [{loop + 1}/{loops}] segment(train) [{(seg_idx + 1)}/{(len(data_samples) - 1) // segment_size + 1}]", logger="current", )