Update test_numeric_batchnorm.py

vacancy / Synchronized-BatchNorm-PyTorch

Synchronized Batch Normalization implementation in PyTorch.

MIT License

1.5k stars 189 forks source link

Closed d-li14 closed 6 years ago

d-li14 commented 6 years ago

Fix nn.BatchNorm2d API incompatible with 2d tensor, tested with PyTorch version 0.4