Based on feedback from @olineumann: Currently, one can only do random guessing on the number of epochs, batch size, ... because there is no feedback on how the training performs. Therefore, as a user of PyWATTS, it would be useful to be able to review the training process of pyTorch models, for example, in terms of a train/test loss curve.
Describe the solution you'd like
Maybe there is a way to add terminal printing? Maybe there is a way to plot train/test loss curves? Maybe there is a way to pass an individual method for evaluation after each epoch?
What is your feature request related to?
Please describe the module/problem
Based on feedback from @olineumann: Currently, one can only do random guessing on the number of epochs, batch size, ... because there is no feedback on how the training performs. Therefore, as a user of PyWATTS, it would be useful to be able to review the training process of pyTorch models, for example, in terms of a train/test loss curve.
Describe the solution you'd like
Maybe there is a way to add terminal printing? Maybe there is a way to plot train/test loss curves? Maybe there is a way to pass an individual method for evaluation after each epoch?