Host data on data.bris.ac.uk (#3)

* Update pretrained model URLs to Data.Bris * [test] Support changing torch hub id via TORCH_HUB env var
epic-kitchens · Aug 5, 2019 · cfc59f1 · cfc59f1
1 parent 4a396f8
commit cfc59f1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pretrained_settings.py b/pretrained_settings.py
@@ -8,7 +8,7 @@
     ["variant", "base_model", "modality", "num_segments", "consensus_type"],
 )
 _epic_url_base = (
-    "https://wp-research-public.s3-eu-west-1.amazonaws.com/epic-models-checkpoints/"
+    "https://data.bris.ac.uk/datasets/2tw6gdvmfj3f12papdy24flvmo/"
 )
 
 
@@ -47,4 +47,4 @@
 
 
 class InvalidPretrainError(Exception):
-    pass
+    pass
diff --git a/tests/test_hub.py b/tests/test_hub.py
@@ -3,9 +3,10 @@
 from tsm import TSM
 from itertools import product
 import pytest
+import os
 
 
-repo = "epic-kitchens/action-models:pytorch-hub-support"
+repo = os.getenv("HUB_ID", "epic-kitchens/action-models:pytorch-hub-support")
 epic_class_counts = (125, 352)
 segment_count = 8