I follow the video below to train the cat_dog model.
Jetson AI Fundamentals - S3E2 - Image Classification Inference
Although I successfully trained the model (model_best.pth.tar is saved and new timestamp is updated ), the accuracy of identifying dog pictures is lower than 20% (by using the photos in “test” folder), but it’s almost 90% for cat photos. I train the model for 70 epochs and got a Acc@1 79% accuracy.
I get an “Segmentation fault (core dumped)” after the model is trained but according to the this threat ( Segmentation fault (core dumped) on jetson nano when training resnet-18 on my small dataset of just 60 images using transfer learning!), it’s fine because the “segfault” happens after
Would you please tell me what’s to do next?
p.s. I also convert model to ONNX successfully with new timestamp updated (Segmentation fault also occurred while converting)