search

neeraja1504 / distriLLM

2 stars 0 forks source link

Make the code modular #1

Open neeraja1504 opened 1 year ago

neeraja1504 commented 1 year ago

1) Datacreation.py -> creation of the data 2) Dataset.py -> dataset and dataloader function 3) Model.py -> Model definition 4) Train.py -> Training loop 5) Test.py -> Evaluation

AdityaKane2001 commented 1 year ago

Make a PR when making changes, for all issues. Self-approve the PR, but mention "Closes #" in the description. See https://github.com/keras-team/keras-cv/pull/505 for usage example.

Edit: You might also want to see the above issue for the PR merge process. It is interesting to see how much thought goes into merging a small chunk of code, since the core functionality was less than 50 lines of code :). Lmk if you have any issues.