After training with multiple gpus with sync_bn on, I would like to do inference on CPU, However, raise the error:
ValueError: SyncBatchNorm expected input tensor to be on GPU
Is there any way to solve this issue ?  Thanks