Skip to content

Commit

Permalink
Refactor /examples/pytorch/NanoGPT/*_train.py (#1111)
Browse files Browse the repository at this point in the history 
* Refactored examples/pytorch/NanoGPT/

* Refactored examples/pytorch/NanoGPT/

* Refactored examples/pytorch/NanoGPT/

* Refactored examples/pytorch/NanoGPT/ds_train.py

* Refactored examples/pytorch/NanoGPT/fsdp_train.py

* Refactor /examples/pytorch/NanoGPT/*_train.py

* Commit pre-commit changes.

* Fix a little bug in examples/pytorch/nanogpt/fsdp_train.py

* Change the "use_native" case in train.py, when loading data.
  • Loading branch information
@Filtee
Filtee committed May 13, 2024
1 parent f6e7059 commit 83047b2
Show file tree
Hide file tree
Showing 3 changed files with 718 additions and 456 deletions.
Loading

0 comments on commit 83047b2

Please sign in to comment.