Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
jfzhang95 committed Dec 1, 2018
1 parent 53efec6 commit b7cbf5a
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions modeling/backbone/xception.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def forward(self, inp):
else:
skip = inp

x += skip
x = x + skip

return x

Expand Down Expand Up @@ -120,7 +120,7 @@ def __init__(self, output_stride, BatchNorm,
self.bn2 = BatchNorm(64)

self.block1 = Block(64, 128, reps=2, stride=2, BatchNorm=BatchNorm, start_with_relu=False)
self.block2 = Block(128, 256, reps=2, stride=2, BatchNorm=BatchNorm, start_with_relu=True,
self.block2 = Block(128, 256, reps=2, stride=2, BatchNorm=BatchNorm, start_with_relu=False,
grow_first=True)
self.block3 = Block(256, 728, reps=2, stride=entry_block3_stride, BatchNorm=BatchNorm,
start_with_relu=True, grow_first=True, is_last=True)
Expand Down Expand Up @@ -190,6 +190,8 @@ def forward(self, x):
x = self.relu(x)

x = self.block1(x)
# add relu here
x = self.relu(x)
low_level_feat = x
x = self.block2(x)
x = self.block3(x)
Expand Down Expand Up @@ -241,6 +243,7 @@ def _init_weight(self):
m.weight.data.fill_(1)
m.bias.data.zero_()


def _load_pretrained_model(self):
pretrain_dict = model_zoo.load_url('http://data.lip6.fr/cadene/pretrainedmodels/xception-b5690688.pth')
model_dict = {}
Expand Down

0 comments on commit b7cbf5a

Please sign in to comment.