diff --git a/torchvision/models/googlenet.py b/torchvision/models/googlenet.py index 1dc5136d726..1d8ecbcd3b2 100644 --- a/torchvision/models/googlenet.py +++ b/torchvision/models/googlenet.py @@ -171,7 +171,7 @@ def eager_outputs(self, x: Tensor, aux2: Tensor, aux1: Optional[Tensor]) -> Goog def forward(self, x: Tensor) -> GoogLeNetOutputs: x = self._transform_input(x) - x, aux1, aux2 = self._forward(x) + x, aux2, aux1 = self._forward(x) aux_defined = self.training and self.aux_logits if torch.jit.is_scripting(): if not aux_defined: