Refactor train/predict function

nmheim / torsk

An echo state network (ESN) for video prediction

10 stars 5 forks source link

Closed nmheim closed 5 years ago

nmheim commented 5 years ago

For netter usability and analysis of time spent

nmheim commented 5 years ago

remember to enable eval mode to increase performance