Merge branch 'fix/general_small_fixes' into 'main'
Fix a number of issues with the infrastructure, no major rework See merge request !11
No related branches found
No related tags found
Showing
- README.md 6 additions, 1 deletionREADME.md
- optimus/dataloader.py 23 additions, 7 deletionsoptimus/dataloader.py
- optimus/datasets/tinystories.py 3 additions, 3 deletionsoptimus/datasets/tinystories.py
- optimus/datasets/wikitext103.py 3 additions, 3 deletionsoptimus/datasets/wikitext103.py
- optimus/example_training.py 3 additions, 9 deletionsoptimus/example_training.py
- optimus/model.py 27 additions, 14 deletionsoptimus/model.py
- optimus/tokenizer.py 4 additions, 4 deletionsoptimus/tokenizer.py
- optimus/trainer.py 11 additions, 13 deletionsoptimus/trainer.py
Loading
Please register or sign in to comment