Atopis

Thank you for your help, it works well on my dataset. And do you know how to turn .pth to .onnx? Thank you!