sm-1z / catface Goto Github PK

hello, 我本来想运行一下你的train.py，我发现你的train和test文件夹名字应该是写反了，因为test文件夹下的图片远远多于train文件夹。
而且我报了这个错误，请问你用这个代码训练的时候遇到过吗？维度不同
Using a target size (torch.Size([10, 16])) that is different to the input size (torch.Size([10, 18])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
‘’’
RuntimeError: The size of tensor a (18) must match the size of tensor b (16) at non-singleton dimension 1

如果方便的话，可不可以上传一下你的预训练模型啊？我只是最近在研究猫的分类，想试试看把猫脸检测出来会不会效果更好一点。
如果不方便也没关系，谢谢你。