Skip to content

add train module and data preparation module#2

Open
amirhossein-yousefi wants to merge 3 commits intoirfanICMLL:mainfrom
amirhossein-yousefi:main
Open

add train module and data preparation module#2
amirhossein-yousefi wants to merge 3 commits intoirfanICMLL:mainfrom
amirhossein-yousefi:main

Conversation

@amirhossein-yousefi
Copy link

In this project, I did the following:

  1. 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.
  2. Adding a train module so you can train this model on any custom dataset.
  3. 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.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant