Fix reference to old main class
This commit is contained in:
parent
d4da4ca121
commit
525298f761
1 changed files with 1 additions and 1 deletions
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue