Add ATNs for ImageNet dataset #87
Labels
Core
Core developer should use this tag
good first issue
Good for newcomers
Modification
Modify some features
Priority:Medium
Medium Priority
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.
The text was updated successfully, but these errors were encountered: