generate text in batch

2021-10-13 00:05:06 +03:00 · 2020-04-27 00:09:03 -04:00
parent aef87eed94
commit 945c72f829
7 changed files with 17 additions and 17 deletions
--- a/textattack/goal_functions/goal_function.py
+++ b/textattack/goal_functions/goal_function.py
@@ -76,7 +76,6 @@ class GoalFunction:
            model_device = next(self.model.model.parameters()).device
        else:
            model_device = next(self.model.parameters()).device
-        print('ids:', ids)
        ids = torch.tensor(ids).to(model_device) 
        #
        # shape of `ids` is (n, m, d)