divyam3897 / ANP_VS

Code for the paper "Adversarial Neural Pruning with Latent Vulnerability Suppression"
http://proceedings.mlr.press/v119/madaan20a/madaan20a.pdf
MIT License
14 stars 1 forks source link

Inconsistencies reproducing the experiments and troubles with CPU/GPU support with suggested requirements #1

Closed giorgiopiras closed 1 year ago

giorgiopiras commented 1 year ago

Problem Introduction

HI @divyam3897 , I am trying to reproduce the network that you used in the experiments of the paper "Adversarial Neural Pruning with Latent Vulnerability Suppression". Specifically, I am trying to train a VGG-16 on CIFAR-10 with your exact same parameters and specifications. When running with Tensorflow 1.14, cudnn 7.6.5, Python 3.5 I yet face several warnings.

Warnings

Comments

I am just reporting some of the Warnings, but messages of this kind are repeated for many operations within the log. Anyway, all these issues possibly lead to Warnings of this kind, which I believe are those describing the issue more accurately: 2022-11-22 07:55:42.072517: I tensorflow/core/platform/cpu_feature_guard.cc:145] This TensorFlow binary is optimized with Intel(R) MKL-DNN to use the following CPU instructions in performance critical operations: SSE4.1 SSE4.2 AVX AVX2 AVX512F FMA. To enable them in non-MKL-DNN operations, rebuild TensorFlow with the appropriate compiler flags.

Eventually, this leads to running the training on the CPU instead of GPU which of course takes ages. I wonder whether it was possible to gently ask for the pretrained model used in the paper, or at least the exact list of requirements needed. I am running on NVIDIA RTX A600, CUDA 11.4. You will find attached my used libraries and the complete log.

Attachments

FULL LOG.txt anpvs_3.txt

divyam3897 commented 1 year ago

Hi,

Thank you for raising this issue. The warnings are most likely due to different versions. I have committed the requirements.txt, which should fix the issue.

giorgiopiras commented 1 year ago

Hi,

Thank you for raising this issue. The warnings are most likely due to different versions. I have committed the requirements.txt, which should fix the issue.

Hi, thanks for your reply. Just for clarity, in case anyone else would need to reproduce the experiments, you also need cudatoolkit=9.0 and matplotlib. It works fine now, thanks.