Added bash scripts

Signed-off-by: Chanakya <[email protected]>
microsoft · Jun 22, 2023 · fa29cb2 · fa29cb2
1 parent a57f9fe
commit fa29cb2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/download_datasets.py b/src/download_datasets.py
@@ -1,4 +1,4 @@
 import gdown
 
-data_url = "https://drive.google.com/drive/folders/1ku9rHvuCIFM_PVzJALcOMgEfgauhkV_f"
+data_url = "https://drive.google.com/drive/folders/1JExEAVW8FIJrD7eKUSSXTzrmyzH0EYkG"
 gdown.download_folder(data_url)
diff --git a/src/main.py b/src/main.py
@@ -141,11 +141,12 @@ def set_seed(seed):
 
     args = parser.parse_args()
 
-    args.feat_dims = 2048  # Where is this used?
+    args.feat_dims = 2048
     args.beta = float(args.beta)
     args.alphas = list(map(float, args.alphas.split(","))) if args.alphas else None
     device = torch.device(f"cuda:{args.cuda}" if torch.cuda.is_available() else "cpu")
 
+    set_seed(args.seed)
     model = get_model(args.net)
     dataset = Dataset(
         dataset_name=args.dataset,