Merge pull request #102 from QData/checkpoint-bug

fix parallel script
QData · May 20, 2020 · 1534e6e · 1534e6e
2 parents 7ba2bd2 + 88e950b
commit 1534e6e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/textattack/shared/scripts/run_attack_parallel.py b/textattack/shared/scripts/run_attack_parallel.py
@@ -50,6 +50,9 @@ def run(args):
         num_examples = resume_checkpoint.num_remaining_attack
         logger.info('Recovered from previously saved checkpoint at {}'.format(resume_checkpoint.datetime))
         print(resume_checkpoint, '\n')
+    else:
+        num_examples_offset = args.num_examples_offset
+        num_examples = args.num_examples
 
     # This makes `args` a namespace that's sharable between processes.
     # We could do the same thing with the model, but it's actually faster