读取模型出现问题，训练模型是.th后缀

您好作者，是不是因为我的训练模型是.th后缀（因为我看作者您的预训练模型是.pth），所以出现以下问题。

args参数没有，参数是不是没有写，前面的代码先人为指定下args看看