From b27b7cc54f450bb5fe8c9ea2faf5e01d0082889a Mon Sep 17 00:00:00 2001 From: Francois Fleuret 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.39.5