PyTorch 1.7.0 Compatibility Updates (#1233)
* torch 1.7.0 compatibility updates * add inference verification
This commit is contained in:
1 parent
453acdec67
commit
c8c5ef36c9
4 files changed
+16
-2
No files matched your search
@@ -74,7 +74,7 @@ def initialize_weights(model):
|
||||
elif t is nn.BatchNorm2d:
|
||||
m.eps = 1e-3
|
||||
m.momentum = 0.03
|
||||
elif t in [nn.LeakyReLU, nn.ReLU, nn.ReLU6]:
|
||||
elif t in [nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6]:
|
||||
m.inplace = True
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user