Skip to content
Snippets Groups Projects

Remove token_line and token_pg from GlobalHTADecoder

Merged Remove token_line and token_pg from GlobalHTADecoder
remove-token-line into main
Merged Mélodie Boillet requested to merge remove-token-line into main
Compare
and
1 file
+ 0
− 2
Compare changes
  • Side-by-side
  • Inline
dan/decoder.py
+ 0
− 2
@@ -381,8 +381,6 @@ class GlobalHTADecoder(Module):
cache=None,
num_pred=None,
keep_all_weights=False,
token_line=None,
token_pg=None,
):
device = raw_features_1d.device