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:
|
if PLOT:
|
||||||
# Plot training success rate (with 'average loss')
|
# Plot training success rate (with 'average loss')
|
||||||
pywatts.main.plot_training(train_eval)
|
pywatts.routines.plot_training(train_eval)
|
||||||
|
|
||||||
|
|
||||||
exit()
|
exit()
|
||||||
|
|
Loading…
Reference in a new issue