checkpoint
SuperTizzyt opened this issue · 1 comments
SuperTizzyt commented
The checkpoint of the model seems to fail, and the predicted category is the same for all images, how can I solve this problem?
csy530216 commented
Hello, I am so sorry for the late reply. If you still need the help, can you explain more about the problem you faced? If the predicted category is the same for all images, perhaps you could check whether nan occurs in the data or computation.