求助compression_imagenet.py实验环境

jiaxiaojunQAQ / Comdefend

The code for ComDefend: An Efficient Image Compression Model to Defend Adversarial Examples (CVPR2019)

112 stars 25 forks source link

Open lidanxu opened 4 years ago

lidanxu commented 4 years ago

作者您好,我下载了您的代码以及纯tensorflow实现的代码,发现这个模型很吃显存,我的环境是但是只有把batch size设置为1才能跑,否则就超内存,可以请问你们实验环境是怎么样的,batch_size=1学不起来但是我个人感觉这个模型没有那么复杂,层数也不多,不知道为何这么吃显存

lidanxu commented 4 years ago

希望您能给我一些帮助,我没搞明白为什么模型跑起来需要那么大的显存这样我本地实验根本做不起来

lidanxu commented 4 years ago

万分感谢!