torchwrench.hub.paths module¶

torchwrench.hub.paths.get_cache_dir(mkdir: bool = False, make_parents: bool = True) → Path[source]¶

Returns torchwrench cache directory for storing checkpoints, data and models.

Defaults is ~/.cache/torchwrench. Can be overriden with ‘TORCHWRENCH_CACHEDIR’ environment variable.

torchwrench.hub.paths.get_tmp_dir(mkdir: bool = False, make_parents: bool = True) → Path[source]¶

Returns torchwrench temporary directory.

Defaults is /tmp/torchwrench. Can be overriden with ‘TORCHWRENCH_TMPDIR’ environment variable.