search

SKKU-ESLAB / Auto-Compression

Automatic DNN compression tool with various model compression and neural architecture search techniques
MIT License
20 stars 20 forks source link

[Question] DNAS-compression have too many memory #46

Open nugabom opened 1 year ago

nugabom commented 1 year ago

I try to apply your DNAS-compression on my project. I have TITAN V to train my model. when I train my model at cifar data, GPU memory is normal at 4000MIB. but when I try to use your DNAS-compression, I encounter memory isssue (i,e. CUDA out of memory) I hope to fix this issue