search

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.42k stars 1.4k forks source link

Allow Configurable Cache Directory #1821

Closed leimao closed 4 months ago

leimao commented 4 months ago

Apex ASP will generate a cache file called master_list.pkl at the root of the working directory and it just looks abrupt.

Change the default behavior to creating a cache directory .cache at the root of the working directory and saving/loading the master_list.pkl there.

The user can also change the cache directory by setting the environment variable ASP_CACHE_DIR in the OS.