Update.

[pytorch.git] / attentiontoy1d.py

diff --git a/attentiontoy1d.py b/attentiontoy1d.py
index ad0c0b1..1dbd614 100755 (executable)
--- a/attentiontoy1d.py
+++ b/attentiontoy1d.py
@@ -12,7 +12,7 @@ from torch.nn import functional as F
 
 ######################################################################
 
-parser = argparse.ArgumentParser(description='Toy RNN.')
+parser = argparse.ArgumentParser(description='Toy attention model.')
 
 parser.add_argument('--nb_epochs',
                     type = int, default = 250)
@@ -147,7 +147,6 @@ def generate_sequences(nb):
 ######################################################################
 
 import matplotlib.pyplot as plt
-import matplotlib.collections as mc
 
 def save_sequence_images(filename, sequences, tr = None, bx = None):
     fig = plt.figure()