Add model definition to test imagenet accuracy