Potential performance optimizations in the train subroutine

BerkeleyLab / inference-engine

A deep learning library for use in high-performance computing applications in modern Fortran

Other

25 stars 6 forks source link

Open rouson opened 1 month ago

rouson commented 1 month ago

[ ] Hoist the if (.not. allocated(...)) ... blocks into an initialization procedure and instead assert allocated status
[ ] Have caller provide a cost array that is of size num_mini_batches (make cost intent(inout))
[ ] Allocate pair_cost array of size maxval(size(mini_batches))