mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1 from cclauss/patch-1

Browse Source 
Identity is not the same thing as equality in Python
This commit is contained in:
Ross Wightman 2019-05-14 08:52:37 -07:00 committed by GitHub
commit 0a84dd5890
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models/dpn.py
View File

@ -167,14 +167,14 @@ class DualPathBlock(nn.Module):
self.num_1x1_c = num_1x1_c self.num_1x1_c = num_1x1_c
self.inc = inc self.inc = inc
self.b = b self.b = b
if block_type is 'proj': if block_type == 'proj':
self.key_stride = 1 self.key_stride = 1
self.has_proj = True self.has_proj = True
elif block_type is 'down': elif block_type == 'down':
self.key_stride = 2 self.key_stride = 2
self.has_proj = True self.has_proj = True
else: else:
assert block_type is 'normal' assert block_type == 'normal'
self.key_stride = 1 self.key_stride = 1
self.has_proj = False self.has_proj = False