"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "427ca4e3f5ac7e83484bfd2e86c11d16d7382dfe"
Commit 2d124b74 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

set batch_size=1 in kl quant

parent d57049bc
......@@ -137,7 +137,8 @@ def main(config, device, logger, vdl_writer):
sample_generator=sample_generator(train_dataloader),
save_model_filename='inference.pdmodel',
save_params_filename='inference.pdiparams',
batch_nums=10)
batch_size=1,
batch_nums=None)
if __name__ == '__main__':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment