search

inspire-group / hydra

Code and checkpoints of compressed networks for the paper titled "HYDRA: Pruning Adversarially Robust Neural Networks" (NeurIPS 2020) (https://arxiv.org/abs/2002.10509).
http://vsehwag.github.io/hydra
90 stars 24 forks source link

Runtime Error #3

Closed 3droot closed 2 years ago

3droot commented 2 years ago

Hello When I run train.py, I encounter CUDA out of memory. that error rises in layers.py, line 74 How can I resolve that problem?