revert demo

This commit is contained in:
gauthiier 2022-02-10 13:39:26 +01:00
parent 9b0a4b5720
commit d62e500c17

View File

@ -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__':