search

BerkeleyLab / inference-engine

A deep learning library for use in high-performance computing applications in modern Fortran
Other
25 stars 6 forks source link

Concurrent loop through each mini-batch during training #178

Closed rouson closed 2 months ago

rouson commented 2 months ago

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