Is your feature request related to a problem? Please describe.
In order to prove that our network works for multiple different models we might want to parse and run resnet18, which uses the same 1000 guess options that alexnet does, but is supposed to be better at guessing than alexnet is.
Describe the solution you'd like
A test, that uses the same code as the imagenet test for alexnet, but that runs resnet18 instead. A resnet18 that is also fetched from Pytorch.
Describe alternatives you've considered
NaN
Additional context
NaN