Correct Plot import

1 files changed, 1 insertions, 1 deletions

diff --git a/train.py b/train.py
index aeb4177..1b75138 100755
--- a/train.py
+++ b/train.py
@@ -16,7 +16,7 @@ from blocks.extensions import Printing, FinishAfter, SimpleExtension
 from blocks.extensions.monitoring import DataStreamMonitoring, TrainingDataMonitoring
 
 try:
-    from blocks.extensions.plot import Plot
+    from blocks.extras.extensions.plotting import Plot
     use_plot = True
 except ImportError:
     use_plot = False