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