AdvancedPhotonSource / tike

Repository for ptychography software
http://tike.readthedocs.io
Other
29 stars 15 forks source link

REF: Move minibatch selection into solvers #123

Closed carterbox closed 3 years ago

carterbox commented 3 years ago

Purpose

Make unique minibatches each epoch instead of using the same minibatches each epoch.

Approach

Move minibatch selection into solvers so that indexing is used to select the minibatches each epoch instead of separating minibatches into a list when the data is pushed to the GPU(s).

No new functionality that needs testing.

Pre-Merge Checklists

Submitter

Reviewer