New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add train module and data preparation module #2

Open

amirhossein-yousefi wants to merge 3 commits into irfanICMLL:main from amirhossein-yousefi:main

amirhossein-yousefi commented Nov 13, 2022

In this project, I did the following:

Adding modules for data preparation for training and writing custom PyTorch dataset class for the CMPFacade dataset so as to iterate through the data efficiently and cleanly.
Adding a train module so you can train this model on any custom dataset.
In the train module I used gradient accumulation as a feature of PyTorch to train the model instead of explicit batch size. It is great when you don't have a powerful GPU.

amirhossein-yousefi added 3 commits

November 13, 2022 14:45


          add files

303323e


          add_more_doc

b3c52b4


          add_more_doc

c70df0d

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment