PyTorch 1.7.0 Compatibility Updates (#1233)

* torch 1.7.0 compatibility updates

* add inference verification
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-10-28 15:03:50 +01:00
1 parent 453acdec67
commit c8c5ef36c9
4 files changed
+16 -2

No files matched your search

+1 -1
View File
@@ -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