dan/decoder.py

@@ -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
 

dan/manager/dataset.py

@@ -232,7 +232,6 @@ class GenericDataset(Dataset):
             self.params["config"]["padding_value"] = self.padding_value
 
         self.curriculum_config = None
-        self.training_info = None
 
     def __len__(self):
         return len(self.samples)