Skip to content

Concurrent loop through each mini-batch during training - #178

Merged
rouson merged 3 commits into
mainfrom
concurrent-training
Jul 18, 2024
Merged

Concurrent loop through each mini-batch during training#178
rouson merged 3 commits into
mainfrom
concurrent-training

Conversation

@rouson

@rousonrouson commented Jul 18, 2024

Copy link
Copy Markdown
Contributor

This PR lays the groundwork for parallelizing or offloading a significant part of the training algorithm to a GPU.

rouson added 3 commits July 17, 2024 22:06
This commit allocates a pair_cost array for each mini-batch so
that each input/output pair's contribution to the cost function
sum can be stored in a separate element rather than keeping a
sequential running tally as the loop through the mini-batch
progresses. This change lays the groundwork for making the
loop concurrent, which in turn lays the foundation for using
a Fortran 2023 concurrent reduction. The calulaiton is currently
redundant with a running sum so that a match between the two can
be verified in an assertion. The reundancy and the assertion will
be removed in a future commit.
@rouson
rouson merged commit 5425151 into mainJul 18, 2024
@rouson
rouson deleted the concurrent-training branch July 18, 2024 05:47
Sign up for freeto 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

@rouson