Update gitignore with server configuration files & change config to run code on GPU (#24)

Author: yashkant

.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

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/
 
+

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

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"]