New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

是用自己的数据集训练报错 #32

Closed

nvliajia opened this issue Feb 23, 2022 · 1 comment

nvliajia commented Feb 23, 2022

用作者的数据集训练可以，使用自己的数据集结果报错，
ValueError: Expected input batch_size (304) to match target batch_size (336).

我把自己的数据集融合到作者的数据集中也能训练，唯独无法单独训练自己的，不知道是不是还应该修改其他参数？

Owner

gitabtion commented Mar 27, 2022

这套代码数据处理部分有点粗糙，兼容性可能差点，有需要的话您自己改改哈

gitabtion closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment