fuqiang

Hello, i resolved the problem by changing the code to ``` model = MobileFaceNet(embedding_size).to(device) head = Arcface(embedding_size=embedding_size, classnum=6056).to(device) head_params = [param for name, param in head.named_parameters()] # Manually create parameter...

well i finally converted model to trt but gives me this error: [TensorRT] ERROR: 4: (Unnamed Layer* 325) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (3,3) with...

also after converting model with torch2trt, model's accuracy drops heavily. why??!!