Skip to content

Commit f21df39

Browse files
committed
remove call to upsample
1 parent 09bafae commit f21df39

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

torchxrayvision/baseline_models/chexpert/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@ def __init__(self, weights_zip="", num_models=30):
5757

5858
def forward(self, x):
5959
x = x.repeat(1, 3, 1, 1)
60-
x = self.upsample(x)
60+
61+
x = utils.fix_resolution(x, 320, self)
62+
utils.warn_normalization(x)
6163

6264
# expecting values between [-1024,1024]
6365
x = x / 512
@@ -79,6 +81,7 @@ def forward(self, x):
7981

8082
def features(self, x):
8183
x = x.repeat(1, 3, 1, 1)
84+
8285
x = utils.fix_resolution(x, 320, self)
8386
utils.warn_normalization(x)
8487

0 commit comments

Comments
 (0)