Browse Source

[ENH] add filter pseudo and remove loss in log

pull/1/head
Tony-HYX 2 years ago
parent
commit
87dd325b8d
1 changed files with 3 additions and 4 deletions
  1. +3
    -4
      abl/bridge/simple_bridge.py

+ 3
- 4
abl/bridge/simple_bridge.py View File

@@ -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",
)



Loading…
Cancel
Save