zincware / ZnNL

A Python package for studying neural learning
Eclipse Public License 2.0
6 stars 1 forks source link

Konsti update loss a res #82

Closed KonstiNik closed 1 year ago

KonstiNik commented 1 year ago

Updating the loss aware reservoir training strategy.

Adding the option of latest_points, which are appended to each batch in the training. The reservoir also only carries indices rather than the actual data, which makes it computationally more efficient. In order to append data to single batches, the _train_epoch method was customized.