seastarmanager

Results 3 comments of seastarmanager

模型训练好了，如何用一张图片去调用预测结果

如果是通过文件的话，你可以自己仿照my_dataset.py里面的get_xxxx_data_loader，去做一个自己的dataloader，就类似于predict的时候差不多。 如果是PIL读入的图片的话，可以参考pytorch的api，用transform把图片变灰度图，变尺寸，转成tensor。然后对tensor进行reshape把图片的三维tensor变成模型需要的四维的。 然后仿照captcha_predict.py的代码，把图片的tensor变成Variable，然后调用cnn的模型进行前向传播，然后再把onehot形式的label按4个位置分别取最大值，最后输出4个字符的结果。

[BUG/Help] <在mac上运行chatglm-6b-int4 报错：copying a param with shape torch.Size([12288, 2048]) from checkpoint, the shape in current model is torch.Size([12288, 4096])>

我也遇到同样的问题 Windows11 Python3.9 torch2.0.0 or torch1.13.1 都有问题 Transformers (compiled from latest git source) CPU：AMD R7-6800H

[BUG] 微调训练后， 一些无关的词条也被污染了？

我感觉也是这样，是不是还得多任务训练？