how to export with onnx

vacancy / Synchronized-BatchNorm-PyTorch

Synchronized Batch Normalization implementation in PyTorch.

MIT License

1.5k stars 189 forks source link

Open tongchangD opened 2 years ago

tongchangD commented 2 years ago

How do I output the model code as onnx

some error export like tihs

RuntimeError: Unsupported: ONNX export of batch_norm for unknown channel size.