Update gitignore with server configuration files & change config to r…

…un code on GPU (#24)
Cloud-CV · Sep 17, 2019 · 74e84fd · 74e84fd
1 parent 03201c8
commit 74e84fd
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,6 +7,11 @@ viscap/checkpoints/
 viscap/data/
 env/
 static/
+chat.ini
+server-log.txt
+uwsgi_params
+visual_dialog_nginx.conf
+
 
 *.pyc
 db.sqlite3

diff --git a/README.md b/README.md
@@ -187,3 +187,4 @@ BSD
 [23]: https://s3.amazonaws.com/visual-dialog/data/v1.0/2019/features_mask_rcnn_x101_test.h5
 [24]: https://rishabhjain2018.github.io/
 
+
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+
 Django==1.11.23
 channels==0.17.2
 asgi-redis==0.14.1
@@ -20,4 +21,4 @@ torch==1.0.0
 tqdm==4.28.1
 yacs==0.1.6
 opencv-python==4.1.0.25
-torchtext==0.3.1
+torchtext==0.3.1
diff --git a/viscap/visdialch/utils/checkpointing.py b/viscap/visdialch/utils/checkpointing.py
@@ -177,10 +177,6 @@ def load_checkpoint(checkpoint_pthpath, device=None):
                 f"({checkpoint_commit_sha}) at which checkpoint was saved,"
                 " are different. This might affect reproducibility."
             )
-
     # load encoder, decoder, optimizer state_dicts
-    if device == torch.device('cpu'):
-        components = torch.load(checkpoint_pthpath, map_location='cpu')
-    else:
-        components = torch.load(checkpoint_pthpath)
+    components = torch.load(checkpoint_pthpath, map_location=device)
     return components["model"], components["optimizer"]