arcface backbone to tensorrt

[guangdongliang]

Thank for your work!
I want to convert arcface to tensorrt, but nn.BatchNorm1d(feature_dim)) do not support 2D in tensorrt. the error message is " what(): No converter registered for op type: Squeeze " .
In torch/onnx/symbolic.py, 2D tensor will be unsqueeze to 3D first and then squeeze to 2D after BatchNorm.

look forward to your replay! thank you!