Skip to content

Commit 3979978

Browse files
author
Fangchang Ma
committed
minor updates to remove outdated code
1 parent c09977a commit 3979978

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

models.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,10 @@
1-
import os
21
import torch
32
import torch.nn as nn
43
import torch.nn.functional as F
54
import torchvision.models
65
import collections
76
import math
87

9-
oheight, owidth = 228, 304
10-
118
class Unpool(nn.Module):
129
# Unpool: 2*2 unpooling with zero padding
1310
def __init__(self, num_channels, stride=2):
@@ -189,8 +186,6 @@ def __init__(self, layers, decoder, output_size, in_channels=3, pretrained=True)
189186
# define number of intermediate channels
190187
if layers <= 34:
191188
num_channels = 512
192-
# Need to modify owidth for ResNet18 model.
193-
owidth = 912
194189
elif layers >= 50:
195190
num_channels = 2048
196191

0 commit comments

Comments
 (0)