add attention_mask

run_interaction.py

@@ -176,11 +176,11 @@ if __name__ == '__main__':
 
     # local test
 
-    args.task = 'train'
-    args.savedir = 'local_test_train'
-    args.epochs = 10
-    args.lr = 1e-5
-    args.config = './config/config_layer_3.json'
+    # args.task = 'train'
+    # args.savedir = 'local_test_train'
+    # args.epochs = 10
+    # args.lr = 1e-5
+    # args.config = './config/config_layer_3.json'