Fixed ESR gan entry in run_pretrained_models.yaml

Author: TomWildenhain-Microsoft

tests/run_pretrained_models.py

@@ -164,7 +164,7 @@ def download_model(self):
         if not os.path.exists(fpath):
             utils.get_url(url, fpath)
         model_path = os.path.join(dir_name, self.local)
-        if not os.path.exists(model_path):
+        if not os.path.exists(model_path) or self.local == ".":
             if ftype == 'tgz':
                 tar = tarfile.open(fpath)
                 tar.extractall(dir_name)
@@ -260,7 +260,7 @@ def run_test(self, name, backend="caffe2", onnx_file=None, opset=None, extra_ops
         if self.url:
             _, dir_name = self.download_model()
             logger.info("Downloaded to %s", dir_name)
-            model_path = os.path.join(dir_name, self.local)
+            model_path = os.path.join(dir_name, self.local) if self.local != "." else dir_name
         else:
             model_path = self.local
 

tests/run_pretrained_models.yaml

@@ -119,7 +119,7 @@ benchtf-gru:
 esrgan-tf2:
   # url: https://tfhub.dev/captain-pool/esrgan-tf2/1/esrgan-tf2_1.tar.gz
   url: https://github.com/captain-pool/GSOC/releases/download/1.0.0/esrgan.tar.gz
-  model: ersgan
+  model: "."
   model_type: saved_model
   input_get: get_beach
   opset_constraints: