From 0ca6f6d779888f1a9fa2caeb1326814094ed6904 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Tue, 19 May 2020 10:58:25 +0200 Subject: [PATCH] Update. --- attentiontoy1d.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- 2.39.5