提交 2d124b74 编写于 作者: L LDOUBLEV

set batch_size=1 in kl quant

上级 d57049bc
...@@ -137,7 +137,8 @@ def main(config, device, logger, vdl_writer): ...@@ -137,7 +137,8 @@ def main(config, device, logger, vdl_writer):
sample_generator=sample_generator(train_dataloader), sample_generator=sample_generator(train_dataloader),
save_model_filename='inference.pdmodel', save_model_filename='inference.pdmodel',
save_params_filename='inference.pdiparams', save_params_filename='inference.pdiparams',
batch_nums=10) batch_size=1,
batch_nums=None)
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册