X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=attentiontoy1d.py;h=ef203403f93fd2a3b3793e746dcbb3ab5939bcdb;hp=ad0c0b142a15007a1af64f1b4148056ecf78cf59;hb=0ca6f6d779888f1a9fa2caeb1326814094ed6904;hpb=602df813edceaae1b4eb2c69bb5cf0f5823444a7

diff --git a/attentiontoy1d.py b/attentiontoy1d.py
index ad0c0b1..ef20340 100755
--- 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)