Merge pull request #7 from w86763777/main

Great work. I just make it runnable out of the box.
MCG-NJU · Nov 16, 2022 · e694e69 · e694e69
2 parents be8db4b + c2a6673
commit e694e69
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/dataset/Charades-STA/charades_test.json → dataset/Charades_STA/charades_test.json b/dataset/Charades-STA/charades_test.json → dataset/Charades_STA/charades_test.json
diff --git a/dataset/Charades-STA/charades_train.json → dataset/Charades_STA/charades_train.json b/dataset/Charades-STA/charades_train.json → dataset/Charades_STA/charades_train.json
diff --git a/mmn/config/paths_catalog.py b/mmn/config/paths_catalog.py
@@ -2,7 +2,7 @@
 import os
 
 class DatasetCatalog(object):
-    DATA_DIR = "datasets"
+    DATA_DIR = ""
 
     DATASETS = {
         "tacos_train":{

diff --git a/train_net.py b/train_net.py
@@ -138,7 +138,7 @@ def main():
     torch.backends.cudnn.deterministic = True
 
     num_gpus = int(os.environ["WORLD_SIZE"]) if "WORLD_SIZE" in os.environ else 1
-    args.distributed = num_gpus > 1
+    args.distributed = True
 
     if args.distributed:
         torch.cuda.set_device(args.local_rank)