lightweight-neural-architecture-search icon indicating copy to clipboard operation
lightweight-neural-architecture-search copied to clipboard

how much memory was used

Open YiChuangDai opened this issue 2 years ago • 1 comments
trafficstars

in paper said that we can use cpu with small memory source ,but when i run sh tools/dist_search.sh configs/classification/deepmad_29M_224.py 32G memory was used in 10 sec and the job has been aborted.

YiChuangDai avatar May 23 '23 02:05 YiChuangDai

To ensure unified implementation of different methods, our repo first transforms the model description into a PyTorch model before computing relevant scores. As a result, more content is cached during the search process, which is unnecessary for DeepMAD. To reduce the memory usage, the simple way is to change the "nproc=64" to "nproc=32/16/8" in the dist_search.sh . It will cost more search time.

xiuyu-sxy avatar May 24 '23 08:05 xiuyu-sxy