From b6ff9cad79a2aed1fc0becad61291c4661ebaeb5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 24 Dec 2018 14:33:05 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index ecbaa341..6ac63480 100755 --- a/models.py +++ b/models.py @@ -89,7 +89,7 @@ class Upsample(torch.nn.Module): self.mode = mode def forward(self, x): - return torch.nn.functional.interpolate(x, scale_factor=self.scale_factor, mode=self.mode) + return nn.functional.interpolate(x, scale_factor=self.scale_factor, mode=self.mode) class YOLOLayer(nn.Module):