idstcv/ZenNAS

evolution search speed

lyg95 opened this issue · 1 comments

lyg95 commented

The paper says searching cost is 0.5 GPU day with NVIDIA V100 GPU, half precision (FP16), batch size 64.
We use supplied script to search network, and set evolution_max_iter=50000, using V100, FP16, batch 64, the whole process spends much time。
for example, using ./Zen_NAS_ImageNet_latency0.8ms..sh to search 0.8ms model, we take 29 hours.
loop_count=47000/50000, max_score=308.6, min_score=293.333, time=28.213h
loop_count=48000/50000, max_score=308.6, min_score=294.01, time=28.78h
loop_count=49000/50000, max_score=308.6, min_score=294.837, time=29.3295h

in paper, evolutionary iterations is 96000, but using 50000, we still cannot achieve 0.5 GPU day to finish search model.

I wonder is there some problems in code or else