From: Francois Fleuret Date: Tue, 19 May 2020 08:58:25 +0000 (+0200) Subject: Update. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=0ca6f6d779888f1a9fa2caeb1326814094ed6904 Update. --- 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)