Sleepychord/CogLTX

请问巨佬,batch size 如何设置?

liutongyang opened this issue · 1 comments

巨佬您好!关于 batch size 我有一些疑惑。
鉴于本文特有的 buffer 数据结构,做 QA 任务时,初始化 Trainer 不需要传入 dataloader,但是 batch size 是在哪里设置的呢?

已解决。
reasoner_module.py line 162