Python tfhub_cache_dir示例

示例#1

    def __init__(self, cache_dir=None):
        """Creates a resolver that streams tar/gz file content over HTTP.

    Args:
      cache_dir: directory to download and cache modules to.
    """
        self._cache_dir = resolver.tfhub_cache_dir(cache_dir, use_temp=True)

示例#2

文件： resolver_test.py 项目： zoujunrong/hub

 def testCacheDir(self):
   # No cache dir set, None is returned.
   cache_dir = resolver.tfhub_cache_dir()
   self.assertEqual(cache_dir, None)
   # Use temp dir.
   cache_dir = resolver.tfhub_cache_dir(use_temp=True)
   self.assertEquals(cache_dir,
                     os.path.join(tempfile.gettempdir(), "tfhub_modules"))
   # Use override
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/d", cache_dir)
   # Use a flag
   FLAGS.tfhub_cache_dir = "/e"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/e", cache_dir)
   FLAGS.tfhub_cache_dir = ""
   # Use env variable
   os.environ[resolver._TFHUB_CACHE_DIR] = "/f"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/f", cache_dir)
   FLAGS.tfhub_cache_dir = "/e"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/f", cache_dir)
   FLAGS.tfhub_cache_dir = ""
   os.unsetenv(resolver._TFHUB_CACHE_DIR)

示例#3

文件： resolver_test.py 项目： jankim/hub

 def testCacheDir(self):
   # No cache dir set, None is returned.
   cache_dir = resolver.tfhub_cache_dir()
   self.assertEqual(cache_dir, None)
   # Use temp dir.
   cache_dir = resolver.tfhub_cache_dir(use_temp=True)
   self.assertEquals(cache_dir,
                     os.path.join(tempfile.gettempdir(), "tfhub_modules"))
   # Use override
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/d", cache_dir)
   # Use a flag
   FLAGS.tfhub_cache_dir = "/e"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/e", cache_dir)
   FLAGS.tfhub_cache_dir = ""
   # Use env variable
   os.environ[resolver._TFHUB_CACHE_DIR] = "/f"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/f", cache_dir)
   FLAGS.tfhub_cache_dir = "/e"
   cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True)
   self.assertEqual("/f", cache_dir)
   FLAGS.tfhub_cache_dir = ""
   os.unsetenv(resolver._TFHUB_CACHE_DIR)

示例#4

文件： compressed_module_resolver.py 项目： zxye3389/hub

def _module_dir(handle):
    """Returns the directory where to cache the module."""
    cache_dir = resolver.tfhub_cache_dir(use_temp=True)
    return resolver.create_local_module_dir(
        cache_dir,
        hashlib.sha1(handle.encode("utf8")).hexdigest())