Fix reference to old main class

This commit is contained in:
Paul Schaub 2018-08-13 17:17:37 +02:00
parent d4da4ca121
commit 525298f761
Signed by: vanitasvitae
GPG key ID: 62BEE9264BF17311

View file

@ -35,7 +35,7 @@ if TRAIN:
if PLOT:
# Plot training success rate (with 'average loss')
pywatts.main.plot_training(train_eval)
pywatts.routines.plot_training(train_eval)
exit()