small-code-cat
small-code-cat
The dataset I use is minist. I hope someone can help me
Yes, my training results are as good as you, but look at my prediction results are very poor. I posted a screenshot of the prediction on it. 
I don't quite understand. I didn't specify imgsz during training. Why do I need to specify when predicting? Isn't this automatically processed by the framework?
Thank you very much, it helped me a lot, it turns out that the imgsz of the two codes are not the same
By the way, there is another question I want to ask. When using torch.hub.load to load the local classification model, it will report an error when going to predict images.
Thank you very much, this update is very useful, I hope yolov5 can do better and better, cheers!
> @small-code-cat ClassificationModel and SegmentationModel types are not yet fully supported by AutoShape, but you can still load them with PyTorch Hub and pass them torch tensors. Note you must...
There is another question. After I train the model with gpu on Linux, loading the model on macos will report an error. What's the reason? 
I have tried this, and I will report this error.
> @small-code-cat perhaps your paths are incorrect. I would just load as in the default example from ultralytics/yolov5 So is there any way to solve this problem?