download

chxw20 · Feb 18, 2021 · 9ed00cd · 9ed00cd
1 parent 648908e
commit 9ed00cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -134,3 +134,4 @@ dmypy.json
 **/tmp/
 **/*~
 **/data/
+**/saved_models/
diff --git a/BAN_vqa/main.py b/BAN_vqa/main.py
@@ -15,7 +15,7 @@
 from utils import trim_collate
 from dataset import tfidf_from_questions
 
-GPUID = 0
+GPUID = 2
 os.environ["CUDA_VISIBLE_DEVICES"] = str(GPUID)
 
 def parse_args():
@@ -117,4 +117,4 @@ def parse_args():
         train_loader = DataLoader(train_dset, batch_size, shuffle=True, num_workers=1, collate_fn=utils.trim_collate)
         eval_loader = DataLoader(val_dset, batch_size, shuffle=False, num_workers=1, collate_fn=utils.trim_collate)
 
-    train(model, train_loader, eval_loader, args.epochs, args.output, optim, epoch)
+    train(model, train_loader, eval_loader, args.epochs, args.output, optim, epoch)
-Original file line number
+Diff line change
@@ Expand Up / @@ -134,3 +134,4 @@ dmypy.json @@
     **/tmp/
     **/*~
     **/data/
+    **/saved_models/