From d62e500c17c9c1e98161a1d7474bff8a82a19978 Mon Sep 17 00:00:00 2001 From: gauthiier Date: Thu, 10 Feb 2022 13:39:26 +0100 Subject: [PATCH] revert demo --- tokenise+train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tokenise+train.py b/tokenise+train.py index 8b2defb..a5dd7d7 100644 --- a/tokenise+train.py +++ b/tokenise+train.py @@ -128,7 +128,7 @@ def main() -> int: output_dir.mkdir() encode(filepath=args.text, blocksize=args.blocksize, vocabsize=args.vocabsize, ouputdir=output_dir, linebyline=args.line_by_line) - train(ouputdir=output_dir, blocksize=args.blocksize, vocabsize=args.vocabsize, num_steps=args.numsteps, gpu=args.gpu) + train(filepath=args.text, ouputdir=output_dir, blocksize=args.blocksize, vocabsize=args.vocabsize, num_steps=args.numsteps, gpu=args.gpu) if __name__ == '__main__':