Segformer Classification model inference issue

Hi, after converting the segformer’s fan_hybrid_tiny.pth classification model file to onnx format for checking the model inference, I got the low confidence score like less than 1 %. So, i thought to use the exact model’s architecture while converting the pth file to onnx file. but i couldn’t find the exact architecture code. Please let me know where can i find the exact architecture and also any suggestion in improving the accuracy