pytorch-syncbn icon indicating copy to clipboard operation
pytorch-syncbn copied to clipboard

Synchronized Multi-GPU Batch Normalization

Results 6 pytorch-syncbn issues
Sort by recently updated
recently updated
newest added

part of your code: m1 = nn.Sequential( nn.Conv2d(3, 3, 1, 1, bias=False), nn.BatchNorm2d(3), nn.ReLU(inplace=True), nn.Conv2d(3, 3, 1, 1, bias=False), nn.BatchNorm2d(3), ).cuda() torch.manual_seed(123) init_weight(m1) m2 = nn.Sequential( nn.Conv2d(3, 3, 1, 1,...

I follow the installation in readme.md. But still got a problem when running the `test.py` file. My torch version is 1.1.0, cudnn is 7.6.0, on ubuntu18.04. ``` Traceback (most recent...

When running models in mixed prediction, pytorch-syncbn cannot be used as it does not support fp16.

It would be easier to access this package when it's added as a pip package. Thanks.

thanks so much for your code! But I want to use sync3d, what shoud I do?

I run the test.py with the number of gpus > 4 and the gpu memory increase all the time. The speed of calculation becomes slow. Any suggestion?