From bb5eb35db7f13e7a6c5bace76e28057d51320ebe Mon Sep 17 00:00:00 2001 From: pblwk <113490383+pblwk@users.noreply.github.com> Date: Fri, 22 Nov 2024 21:24:20 +0800 Subject: [PATCH] Fixed a bug in googlenet.py Corrected the order of aux1 and aux2. --- torchvision/models/googlenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: