From b27b7cc54f450bb5fe8c9ea2faf5e01d0082889a Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
Date: Tue, 19 May 2020 15:29:26 +0200
Subject: [PATCH] Update.

---
 attentiontoy1d.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/attentiontoy1d.py b/attentiontoy1d.py
index 1dbd614..6540a0f 100755
--- a/attentiontoy1d.py
+++ b/attentiontoy1d.py
@@ -10,6 +10,8 @@ import torch, math, sys, argparse
 from torch import nn
 from torch.nn import functional as F
 
+import matplotlib.pyplot as plt
+
 ######################################################################
 
 parser = argparse.ArgumentParser(description='Toy attention model.')
@@ -146,8 +148,6 @@ def generate_sequences(nb):
 
 ######################################################################
 
-import matplotlib.pyplot as plt
-
 def save_sequence_images(filename, sequences, tr = None, bx = None):
     fig = plt.figure()
     ax = fig.add_subplot(1, 1, 1)
-- 
2.20.1