search

Adversarial-Deep-Learning / code-soup

This is a collection of algorithms and approaches used in the book adversarial deep learning
MIT License
18 stars 18 forks source link

Add ATNs for ImageNet dataset #87

Open gchhablani opened 2 years ago

gchhablani commented 2 years ago

This issue is a continuation of #19. The code for MNIST will be merged in #85. The original paper also mentions architectures for ImageNet models. It would be nice to have these implementations in the repository.

In this comment, I have added dummy code for all architectures needed. However, the implementation should contain exactly the same number of channels/kernel sizes as the original model.

This will need effort in checking the what input/output paddings and strides will work given the architecture.

soham-chitnis10 commented 2 years ago

I would like to take this up