From: Francois Fleuret <francois@fleuret.org>
Date: Fri, 27 Nov 2020 22:13:08 +0000 (+0100)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=de3a0375a79cbbf4299aacc41db1426a39ca9664

Update.
---

diff --git a/attentiontoy1d.py b/attentiontoy1d.py
index e82894e..2cecad8 100755
--- a/attentiontoy1d.py
+++ b/attentiontoy1d.py
@@ -321,6 +321,8 @@ if args.with_attention:
 test_input = test_input.detach().to('cpu')
 test_outputs = test_outputs.detach().to('cpu')
 test_targets = test_targets.detach().to('cpu')
+test_bx = test_bx.detach().to('cpu')
+test_tr = test_tr.detach().to('cpu')
 
 for k in range(15):
     save_sequence_images(