CooperLeong

The paper says they pre-trained the model on Image-net, have you tried pre-training?

https://user-images.githubusercontent.com/5608480/205428787-caed23b8-7a1f-4cc3-8a4c-b94c5e73c647.mp4 I just cut this on my own; see if it works!

Check if your onnxruntime is gpu version?