gauthiier/NATURESPEAK-ML-UTTER
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

revert demo

Browse Source
This commit is contained in:
gauthiier 2022-02-10 14:13:00 +01:00
parent 430f6f91e0
commit 2ea11e0dbd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tokenise+train.py
View File

@ -48,7 +48,7 @@ def train(filepath: str, ouputdir: Path, blocksize: int, vocabsize: int, num_ste
# data = TokenDataset(dat, tokenizer_file=tok, block_size=blocksize, from_cache=True) # data = TokenDataset(dat, tokenizer_file=tok, block_size=blocksize, from_cache=True)
ai.train(filepath, output_dir=str(ouputdir), line_by_line=False, from_cache=False, learning_rate=1e-3, batch_size=128, num_steps=num_steps, generate_every=1000, save_every=1000, num_workers=4) ai.train(filepath, output_dir=str(ouputdir), line_by_line=False, from_cache=False, learning_rate=1e-3, batch_size=256, num_steps=num_steps, generate_every=1000, save_every=1000)
return "Done!" return "Done!"