search

facebookresearch / ImageNet-Adversarial-Training

ImageNet classifier with state-of-the-art adversarial robustness
Other
674 stars 87 forks source link

Running without horovod? #8

Closed lhfowl closed 5 years ago

lhfowl commented 5 years ago

I'm having trouble getting horovod installed properly on my computer. Is there away to predict on the model without horovod?

ppwwyyxx commented 5 years ago

You can use inference-example.py.

You can also remove everything in main.py about horovod and use the hvd.size() == 1 codepath only: https://github.com/facebookresearch/ImageNet-Adversarial-Training/blob/c704ef76467225aba0a69b3c5d04333b7f3f9526/main.py#L201-L204