Skip to content

fix bug: bs=4 ok while bs=32 error#1

Open
TransformersWsz wants to merge 4 commits intoliucongg:mainfrom
TransformersWsz:main
Open

fix bug: bs=4 ok while bs=32 error#1
TransformersWsz wants to merge 4 commits intoliucongg:mainfrom
TransformersWsz:main

Conversation

@TransformersWsz
Copy link
Copy Markdown

训练集总共27768条数据:

  • batch_size=4时可以整除,程序运行正常
  • 当batch_size=32时不可以整除,这时多出来27768%32=24条数据,data[-27768%32:]data[-(27768%32):] 的结果是不一样的,程序会报错

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant