brucefan1983 / GPUMD

Graphics Processing Units Molecular Dynamics
https://gpumd.org/dev
GNU General Public License v3.0
454 stars 115 forks source link

mini-batch training #642

Closed brucefan1983 closed 3 months ago

brucefan1983 commented 3 months ago

This PR aims to improve the training results when using mini-batches, based on the following tricks: