PyTorch inference on tensors of a particular size cause Illegal Instruction (core dumped) on Jetson Nano

Hi there,

I’m trying to run webcam feed based inference on a pretrained pytorch model on the Jetson Nano. When I run a frame through the network I get the

Illegal instruction (core dumped)

error message. I can reliably reproduce this error using the following minimal script:

import torch
import torch.nn as nn

lstm = nn.LSTM(63, 70, 1, True)
x = torch.randn(1,14,63)
lstm(x)

The dimensions of the model and the input are chosen to fit the data I’m using. In reality the model should perform inference on tensors of size (1,175,63). Interestingly running the following:

import torch
import torch.nn as nn

lstm = nn.LSTM(63, 70, 1, True)
for i in range(175):
     x = torch.randn(1,i,63)
     lstm(x)

produces output until the size of the second dimension of x reaches 14 at which point the core dumped error message occurs. Any ideas what’s happening here? I’ve read that torch has some issues with old CPUs but seeing as it seems to be working up to a certain point I doubt this was an issue. I tested this by trying out some solutions posted here which involved entering

export OPENBLAS_CORETYPE=ARMV8

before running the script in the terminal. I’m also running numpy version 1.19.4 which was suggested to be a solution for a similar problem in the above thread so that’s probably not the issue.

Some specs:

python version: 3.6.9
numpy version: 1.19.5
torch version: 1.9.0

system: Jetson Nano 2GB
os: Ubuntu 18.04.5 LTS (Bionic Beaver)

many thanks.

Hi,

First, do you use the prebuilt PyTorch package shared on this topic?
If not, please give it a try.

We test your sample on a Nano 4GB device and it can work well.

$ python3
Python 3.6.9 (default, Jan 26 2021, 15:33:00)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torch.nn as nn
>>> lstm = nn.LSTM(63, 70, 1, True)
>>> x = torch.randn(1,14,63)
>>> lstm(x)
(tensor([[[ 5.6031e-02, -8.8360e-03,  1.0229e-02, -3.7838e-03, -1.0825e-01,
          -1.0935e-02, -2.5602e-02,  5.2157e-02,  1.6666e-02,  7.0142e-02,
          -3.7780e-02, -6.0340e-02,  1.1501e-01, -1.1459e-01, -3.8188e-04,
          -6.5344e-02,  1.3932e-01, -5.5914e-03, -9.8291e-02, -1.4367e-01,
           8.5975e-02,  2.9691e-03, -8.7870e-02, -3.1488e-01, -1.2849e-02,
          -3.7763e-02,  1.8979e-01, -3.9970e-02, -3.7594e-01,  2.1132e-01,
          -1.5544e-01,  3.6806e-01,  5.2226e-02, -2.0016e-01,  1.9568e-01,
          -2.9962e-01, -1.0296e-02, -1.0462e-01,  1.3558e-01,  4.9008e-02,
           3.1331e-02, -1.5108e-02,  8.5912e-03, -1.8172e-01, -5.4976e-02,
          -2.8634e-02,  1.4950e-01,  1.6224e-01, -9.6563e-02,  1.3238e-01,
          -1.7478e-03,  2.0347e-01,  5.5988e-02, -1.5413e-03, -9.1996e-02,
          -1.1723e-01, -2.9780e-02,  2.1634e-01, -1.3447e-02, -1.0505e-03,
           5.1296e-02,  4.4293e-02, -3.7859e-02, -4.8863e-02,  9.2663e-02,
          -8.7870e-02, -1.0053e-01,  2.8272e-03, -4.7203e-02, -1.2656e-01],
         [ 1.0596e-01, -2.3782e-02, -1.9335e-01, -1.4489e-02, -8.9978e-02,
          -2.2302e-03, -5.6742e-02, -1.0794e-01, -1.9884e-02, -2.8014e-02,
          -1.8144e-02, -7.2089e-02, -1.0659e-01, -4.7132e-02, -7.1073e-02,
           2.5898e-01, -6.8692e-02, -5.9335e-02,  1.0095e-01,  2.7785e-03,
          -1.4248e-01, -1.0340e-01, -9.8229e-02,  2.8893e-02, -7.9089e-02,
           1.4736e-01, -2.9456e-03, -5.7306e-02,  1.7827e-01,  1.0001e-01,
           3.2359e-01,  5.4744e-02, -7.5327e-02,  1.1609e-01, -6.7285e-02,
          -2.2617e-02,  1.5121e-01,  1.1619e-01,  4.0007e-02, -1.1279e-01,
          -2.8286e-02, -1.8952e-01,  1.0360e-01,  5.2665e-02, -5.0287e-02,
          -6.0863e-02, -7.5621e-02, -1.7445e-01,  2.9395e-02,  6.6238e-02,
          -8.2340e-03,  3.2121e-02,  8.7107e-02,  3.7411e-02, -7.2104e-02,
          -1.4122e-01,  1.5281e-01,  3.4332e-02,  1.1002e-01,  5.5575e-02,
          -7.9585e-02, -4.7398e-02,  1.5375e-01,  2.4957e-02,  2.0601e-02,
           1.3080e-01, -2.6788e-02, -2.4200e-01,  8.6411e-02,  1.1811e-02],
         [ 4.2684e-02, -2.7031e-03,  1.4298e-02, -1.0974e-01,  1.3387e-01,
          -4.4143e-02, -5.9928e-02, -3.4873e-02,  5.4613e-02, -1.3968e-01,
          -1.0887e-01, -5.5022e-02,  1.9583e-02,  1.2343e-01, -9.9416e-02,
           7.9870e-02,  3.1370e-02, -1.6668e-02, -4.7148e-02,  4.7422e-02,
          -1.4521e-01, -7.0527e-02,  3.6346e-02, -9.6922e-03, -6.4716e-03,
           1.0765e-01,  2.4050e-02, -6.9346e-02,  4.4194e-03, -1.3617e-01,
          -1.4726e-01,  7.6511e-03, -1.4076e-01,  1.6348e-01,  5.5327e-02,
          -1.0936e-01, -4.5788e-04, -8.4764e-02,  2.0963e-01, -1.4858e-01,
           1.3468e-01, -1.6554e-01,  1.9735e-01,  6.8844e-02,  4.0271e-02,
          -1.5831e-01,  1.6706e-01, -3.4029e-02, -8.1691e-02, -3.5286e-02,
          -2.1685e-02,  5.3153e-02, -3.5246e-02,  1.7635e-01, -5.7775e-02,
           1.4218e-01, -3.1825e-02, -3.3248e-03, -1.3808e-02,  2.4296e-01,
          -6.7138e-02, -8.7007e-02, -2.8862e-02, -7.4122e-02,  1.1170e-01,
          -1.2967e-01,  2.0547e-01, -8.3149e-02,  9.3661e-02,  1.1447e-01],
         [ 2.0050e-01, -2.6334e-02,  2.3495e-02,  1.8985e-01,  8.1286e-02,
          -1.8525e-02, -1.4701e-02,  6.1575e-02, -2.6881e-02, -9.7923e-02,
          -3.9525e-02,  1.3943e-02, -2.2108e-01,  8.2307e-02,  5.2262e-02,
          -1.4908e-01,  4.1394e-02,  3.9383e-02, -2.9643e-02,  8.2982e-02,
           1.0812e-01, -8.5415e-02,  1.0080e-01,  1.8355e-02,  9.5582e-02,
           1.0853e-01, -9.6317e-02,  2.3256e-02, -7.9257e-02,  1.3188e-02,
           2.5164e-02, -1.6553e-02,  1.3776e-02, -4.4792e-02, -1.6034e-02,
          -1.7137e-02, -1.6599e-01, -1.1741e-01,  9.7626e-02, -2.0726e-01,
          -6.7859e-02,  9.3904e-02,  5.1553e-02, -1.2293e-02, -7.0319e-02,
          -1.0582e-01,  1.4354e-01, -1.1074e-02, -1.6053e-01,  5.4912e-02,
          -9.8129e-02,  1.7655e-01, -3.9173e-02,  4.7816e-02, -7.5706e-02,
          -3.0862e-02, -3.4565e-02,  1.6844e-01, -2.2667e-01,  7.1861e-02,
          -1.1307e-01, -2.6534e-02,  3.3843e-02, -3.0914e-02,  6.7711e-02,
          -8.4125e-02, -8.5001e-02,  2.6888e-02,  2.5008e-01, -2.2955e-03],
         [ 5.1989e-02, -6.1134e-02, -3.0335e-02,  1.1322e-01, -1.1330e-01,
           3.5291e-02, -2.1876e-02,  6.4551e-03,  4.5225e-02, -1.5405e-01,
           1.2379e-01, -8.2270e-02, -3.0484e-01,  1.0819e-01,  1.3357e-01,
           3.5685e-02, -5.5423e-02,  2.7528e-02, -1.6097e-01,  4.1578e-02,
          -4.5584e-02,  3.7888e-02,  1.2584e-01,  6.4713e-02, -1.0298e-01,
           1.4781e-01, -1.1396e-02,  4.1024e-02, -7.4025e-03,  1.2394e-01,
           4.1668e-02,  4.5705e-03,  1.1440e-01,  6.4562e-02,  1.5776e-01,
           9.3784e-02,  5.3231e-02, -1.4836e-01,  1.1751e-01, -1.3569e-02,
          -2.8206e-01, -5.0432e-02,  1.3207e-01,  9.2000e-02,  1.1093e-01,
           1.1821e-02, -4.3048e-02, -1.0227e-01, -2.2217e-02,  3.5602e-03,
          -2.0588e-02, -4.1376e-02, -3.5228e-02,  1.0398e-01, -2.0375e-01,
          -1.2932e-01,  2.7992e-01,  1.8510e-01, -9.4242e-02, -9.9311e-02,
          -1.2898e-01,  1.2948e-01,  1.0802e-02,  1.8211e-02,  2.0469e-02,
          -4.0870e-02,  1.1680e-02,  1.0431e-01,  4.0065e-02,  3.8654e-02],
         [ 9.9353e-02,  6.4253e-02, -3.4160e-02,  2.7182e-02,  4.1429e-02,
           1.7252e-01, -3.7846e-03,  3.3930e-02, -2.6656e-03, -2.2649e-05,
          -1.1989e-01,  5.7061e-02, -3.8611e-02,  8.7217e-03,  1.0726e-01,
           8.8789e-03,  1.8363e-01,  2.9153e-01, -3.0007e-01,  5.6727e-02,
          -9.9625e-02, -2.1588e-02, -4.1067e-01,  9.8972e-02, -7.5088e-02,
           1.2729e-01,  1.0314e-01, -1.8490e-01,  2.5720e-02,  1.7584e-02,
           5.6193e-03, -2.8768e-02,  3.2220e-02, -1.6139e-01, -1.4008e-02,
          -2.0644e-01, -1.4936e-01, -7.6882e-02,  2.6214e-01,  1.0562e-01,
           2.4997e-02, -9.4328e-03, -8.5398e-02, -5.7442e-02, -8.4854e-02,
           1.2768e-02,  1.0603e-01,  6.0193e-02, -1.3217e-01, -2.2714e-02,
          -2.0082e-01,  2.8143e-02, -1.1461e-01,  1.3467e-01, -1.6298e-01,
          -1.3805e-01, -2.7936e-01,  2.1600e-01,  7.9968e-02, -1.3919e-01,
          -6.6155e-02, -4.8140e-02, -4.2059e-02, -4.0622e-02,  1.4554e-01,
           4.0983e-02,  1.1232e-01,  1.5791e-01, -2.0659e-01,  1.0685e-01],
         [-7.4555e-02, -9.8336e-04,  4.8372e-02, -8.8675e-02,  2.1662e-01,
          -2.5914e-01,  1.1040e-01, -1.7399e-01, -1.1990e-01, -7.3328e-02,
           1.0765e-03,  9.5212e-02,  6.4974e-02,  1.8605e-02, -1.4860e-01,
          -4.2923e-02, -6.6076e-02,  1.3738e-01, -1.2874e-01,  2.2139e-01,
          -1.1609e-01,  2.0101e-01,  1.3427e-01,  1.3271e-01, -7.0087e-02,
          -1.6974e-01, -2.0811e-01, -1.3262e-02, -2.7487e-01, -7.3588e-04,
          -5.0719e-02, -1.2575e-01, -3.1040e-01, -3.2686e-02, -4.4878e-02,
          -1.3627e-01,  4.3030e-02,  3.7402e-02, -9.9832e-02, -2.3791e-01,
           1.1643e-01,  5.1211e-02,  2.7892e-01, -6.7376e-02, -3.3468e-02,
           6.5023e-03, -2.6187e-02,  2.3592e-02, -1.1229e-01,  1.8710e-02,
           1.1554e-01,  4.7669e-02,  1.2471e-01,  1.5595e-02, -8.9394e-02,
           1.0160e-01, -7.0636e-02, -3.6355e-02, -7.2106e-04, -3.2922e-03,
          -3.1203e-01, -8.2269e-02,  1.3747e-01,  2.8943e-01, -2.5065e-01,
          -1.3635e-01, -1.4406e-01, -1.5565e-01, -3.8379e-02, -4.2467e-03],
         [-1.6054e-01, -1.7197e-02, -3.2800e-02, -2.3965e-02, -5.3158e-02,
           1.2333e-02,  8.9618e-03,  6.8235e-02,  9.7215e-02,  1.1321e-02,
           2.4302e-01, -5.5776e-02, -1.1977e-01, -1.0460e-01,  6.6412e-02,
           7.4097e-02,  6.1163e-02, -1.9765e-01,  1.1550e-01,  3.1729e-02,
           1.2110e-01,  3.6793e-02, -2.8560e-01, -5.0376e-02, -1.5311e-01,
          -6.0325e-02, -8.7698e-02,  4.9468e-02, -4.6984e-02,  2.4592e-01,
           2.5259e-02, -1.2948e-01,  1.6054e-01, -1.2765e-01,  1.5064e-01,
          -3.9389e-02, -6.3692e-02, -1.0571e-01, -6.0839e-02,  5.5169e-02,
           5.2784e-02, -1.4439e-01, -2.2454e-02,  6.5641e-02, -1.1294e-02,
           1.6320e-01,  1.6501e-01, -6.6609e-02, -1.6454e-02,  2.5773e-01,
          -1.3573e-01, -9.2846e-02,  5.8761e-03, -8.7776e-02,  3.1023e-02,
          -1.4960e-01, -6.0161e-02, -1.1040e-01,  1.0346e-01, -1.1536e-01,
           1.3904e-01,  1.1988e-01,  1.0675e-01, -6.4604e-02, -1.3922e-01,
           3.0046e-02,  1.1107e-01,  9.9815e-02,  3.8590e-03, -1.1489e-01],
         [ 1.7796e-01,  3.3173e-03,  3.6310e-02, -6.8334e-02, -4.5901e-02,
           1.2242e-01, -5.9630e-02,  6.2765e-02, -1.2207e-01, -1.6675e-01,
           1.0564e-01,  7.8418e-02,  1.1342e-02,  1.2741e-01, -1.5002e-01,
          -1.4442e-01, -5.3747e-02,  5.2537e-02, -6.8010e-02,  2.7846e-02,
          -5.4334e-02, -2.1628e-02,  1.5251e-01, -1.8077e-01,  5.4305e-03,
           8.1400e-02,  2.3313e-01, -1.7239e-01, -2.1218e-01,  9.9754e-02,
          -1.9443e-01,  2.4601e-01,  5.9472e-02, -1.9648e-01, -3.1408e-03,
          -5.7469e-02,  1.4491e-02, -1.4890e-01,  1.9786e-02, -8.8356e-02,
          -6.7325e-02, -5.4330e-02,  6.3219e-02,  7.0149e-05,  2.6016e-02,
           5.6930e-03,  7.0553e-02, -1.0225e-01, -8.5517e-02,  4.9350e-02,
          -1.8276e-01,  2.2137e-02, -6.4787e-02,  1.1811e-01, -1.3203e-01,
          -1.5250e-01,  3.3393e-02,  1.8681e-01, -2.0384e-01, -8.0982e-03,
          -1.6721e-01,  7.6560e-02,  2.3648e-03, -7.9858e-02,  1.3694e-01,
           8.9148e-02, -9.7790e-02, -1.0912e-01, -1.2010e-01, -3.2526e-01],
         [ 1.0796e-02,  1.7112e-01, -8.0794e-02, -1.0201e-02, -1.6678e-01,
           4.8956e-02, -3.3468e-02, -1.1329e-01, -1.4307e-01,  3.8808e-03,
          -1.7007e-01,  1.1081e-01,  3.9384e-02, -8.6496e-02,  1.6035e-02,
           1.3158e-02,  2.2107e-01, -4.7775e-02, -1.9427e-01,  2.0944e-01,
          -7.5677e-02, -1.1798e-01,  1.1025e-01, -1.8711e-01,  3.3253e-02,
          -1.1324e-01,  1.2248e-01, -7.5973e-02,  8.3296e-03, -1.4975e-01,
          -1.0782e-01, -5.1636e-02, -1.3034e-01, -1.0203e-01, -5.1191e-02,
          -7.6037e-02,  3.1529e-02, -1.3401e-01, -7.4482e-02,  2.1677e-01,
           8.4192e-02,  1.0355e-01,  1.0202e-02, -1.0927e-01, -2.2705e-01,
           3.8898e-02,  8.1357e-02,  1.3390e-01,  1.7494e-01, -1.6710e-01,
           9.5442e-02,  8.4279e-02, -2.0884e-01,  2.8360e-01, -2.1652e-02,
           3.0337e-01, -5.0468e-02, -6.0181e-02, -9.9337e-02,  1.5276e-01,
           7.6341e-02,  1.4394e-01,  1.1333e-01,  1.3480e-01,  5.9742e-02,
          -2.8182e-02,  5.9056e-02, -1.5152e-01,  2.1141e-01,  2.2443e-01],
         [ 3.1242e-01, -1.3328e-02, -6.4702e-02, -7.3662e-02,  3.3083e-02,
          -1.8229e-01,  3.9831e-02,  2.8169e-02,  2.9558e-01, -8.1146e-02,
          -5.9358e-02,  1.4800e-01,  6.7160e-02,  5.6184e-02,  6.5894e-02,
           8.5739e-02, -4.2400e-02, -5.4244e-02, -9.8622e-02, -2.8258e-03,
           7.4035e-02, -1.1368e-01, -5.8690e-03, -4.2679e-02,  4.3065e-03,
          -9.3517e-02,  2.3469e-01,  1.0425e-01, -3.9116e-02, -1.1520e-01,
           1.9604e-01,  9.0298e-03, -1.4587e-01, -8.2396e-02,  1.6304e-01,
           9.8176e-02, -1.2156e-01, -1.3441e-01,  1.1860e-01,  1.1502e-01,
          -7.3041e-02,  1.3788e-01, -7.3918e-03,  3.0716e-02,  1.1952e-01,
           1.6556e-01,  4.3813e-02, -1.2945e-01,  1.7765e-01, -1.0628e-01,
           2.0266e-01, -1.4229e-01, -8.0150e-02, -3.0512e-01,  1.6196e-01,
           3.0840e-02,  1.1530e-01, -2.5387e-02, -8.1937e-02,  1.6545e-01,
          -1.4643e-01, -6.8823e-02,  5.6442e-02, -2.6221e-01,  6.0834e-02,
           4.5801e-02,  1.0073e-01, -4.4227e-02,  1.2012e-01, -3.6080e-03],
         [ 2.0685e-01, -1.2266e-02,  3.7424e-02,  1.3822e-01, -7.3270e-02,
           1.1441e-01, -1.2382e-01,  1.2796e-01, -6.4230e-02, -1.5086e-01,
          -1.6847e-02, -1.9173e-01, -6.3245e-02, -3.6363e-02, -3.2275e-02,
           3.1215e-02, -3.0032e-02,  5.2593e-02, -7.6114e-02, -1.1319e-01,
          -4.5392e-02, -1.1279e-03, -1.5751e-01,  1.4057e-02, -1.2208e-02,
           1.8330e-01,  5.4593e-02, -5.1254e-02,  1.5672e-02, -2.0811e-02,
          -4.6908e-02,  1.1040e-01,  3.2581e-03, -6.4124e-02,  9.6250e-02,
          -5.3775e-02,  1.1197e-01, -3.8666e-02, -2.0764e-01,  2.2548e-04,
           2.7937e-02, -1.0908e-01,  3.4961e-01,  1.8435e-02, -6.5997e-02,
           1.2136e-01, -3.5810e-03, -7.4943e-02,  9.9110e-02, -8.0469e-02,
           1.3041e-01, -1.7082e-01,  3.6026e-03, -9.0824e-04,  6.7820e-02,
          -2.5519e-02,  8.1379e-02,  8.5598e-02,  7.6895e-02, -3.6114e-03,
          -5.1131e-02, -7.5736e-03,  1.7272e-02,  6.1835e-02,  1.7413e-01,
          -1.4059e-01,  4.3275e-02,  2.5384e-02,  1.5131e-01,  1.9625e-01],
         [ 1.6178e-01, -7.0462e-02,  5.1837e-02,  1.2976e-01, -1.9617e-01,
          -3.4532e-02,  1.9907e-01,  5.2852e-02, -1.7697e-01,  4.6298e-02,
           2.2926e-01,  8.3025e-02,  1.1260e-01, -1.0937e-01, -5.7699e-02,
           2.3633e-01,  1.5617e-02,  3.5624e-02,  5.5827e-03,  1.3430e-01,
          -2.5599e-03,  2.0441e-01, -5.0584e-02,  1.3386e-01,  1.3817e-01,
           9.8355e-02,  9.8374e-02, -6.2096e-02,  1.0149e-01,  6.8557e-02,
          -2.1888e-01, -1.3535e-01,  1.3701e-01, -1.5107e-01,  1.8643e-01,
          -4.9615e-02,  6.9683e-02,  1.4800e-01, -7.8982e-02, -4.5765e-02,
           5.8402e-02, -6.0298e-02,  4.4689e-02,  2.0854e-02,  5.9750e-02,
           1.0959e-01,  8.6809e-02,  6.2491e-03, -1.4493e-01, -1.5228e-02,
           1.5438e-01, -2.5044e-02, -8.9997e-02,  9.6676e-02,  4.4962e-02,
          -1.1919e-01, -3.1666e-02,  5.8698e-02, -1.8586e-01, -8.2214e-02,
          -2.0919e-02,  6.4359e-02,  1.7290e-02, -1.3625e-01, -9.5965e-03,
          -2.5466e-02,  1.3502e-01,  7.1203e-02, -1.0597e-01, -1.3823e-02],
         [-7.0465e-02, -1.7589e-01, -4.7922e-02, -6.0889e-02,  1.8730e-01,
           4.7207e-02,  1.4650e-01, -1.1668e-01,  6.5600e-02, -2.0100e-02,
           1.7324e-02, -7.2771e-02, -9.2855e-03,  1.0280e-01, -9.2456e-02,
           6.6540e-02, -8.4615e-02, -4.2603e-02,  1.4238e-01,  1.6706e-01,
           1.4694e-01,  7.8026e-02, -2.4749e-02, -6.1624e-02, -3.3867e-01,
          -9.2765e-02, -6.2037e-02, -7.0383e-02,  8.2647e-02, -1.4202e-01,
          -1.3576e-03, -9.3392e-02, -1.0022e-01, -3.3840e-03,  1.5150e-01,
           6.9218e-02,  6.3082e-02,  9.3894e-02, -5.2440e-02,  1.7739e-01,
          -2.6205e-02,  3.1875e-02, -6.3834e-03, -2.5625e-03,  9.3113e-02,
          -1.1761e-01, -1.3483e-01, -8.2253e-02, -2.5895e-01, -2.4529e-05,
          -5.8787e-02, -8.7753e-02, -1.0200e-01, -4.3242e-02,  9.7991e-02,
           1.5287e-01, -7.8606e-02, -7.7933e-02,  3.2688e-02, -9.1445e-02,
           4.1327e-02, -3.6453e-02, -1.6838e-01,  1.1062e-01, -2.5284e-01,
           2.2474e-01,  9.0995e-02, -1.5933e-02,  1.2434e-01, -1.0364e-01]]],
       grad_fn=<StackBackward>), (tensor([[[ 5.6031e-02, -8.8360e-03,  1.0229e-02, -3.7838e-03, -1.0825e-01,
          -1.0935e-02, -2.5602e-02,  5.2157e-02,  1.6666e-02,  7.0142e-02,
          -3.7780e-02, -6.0340e-02,  1.1501e-01, -1.1459e-01, -3.8188e-04,
          -6.5344e-02,  1.3932e-01, -5.5914e-03, -9.8291e-02, -1.4367e-01,
           8.5975e-02,  2.9691e-03, -8.7870e-02, -3.1488e-01, -1.2849e-02,
          -3.7763e-02,  1.8979e-01, -3.9970e-02, -3.7594e-01,  2.1132e-01,
          -1.5544e-01,  3.6806e-01,  5.2226e-02, -2.0016e-01,  1.9568e-01,
          -2.9962e-01, -1.0296e-02, -1.0462e-01,  1.3558e-01,  4.9008e-02,
           3.1331e-02, -1.5108e-02,  8.5912e-03, -1.8172e-01, -5.4976e-02,
          -2.8634e-02,  1.4950e-01,  1.6224e-01, -9.6563e-02,  1.3238e-01,
          -1.7478e-03,  2.0347e-01,  5.5988e-02, -1.5413e-03, -9.1996e-02,
          -1.1723e-01, -2.9780e-02,  2.1634e-01, -1.3447e-02, -1.0505e-03,
           5.1296e-02,  4.4293e-02, -3.7859e-02, -4.8863e-02,  9.2663e-02,
          -8.7870e-02, -1.0053e-01,  2.8272e-03, -4.7203e-02, -1.2656e-01],
         [ 1.0596e-01, -2.3782e-02, -1.9335e-01, -1.4489e-02, -8.9978e-02,
          -2.2302e-03, -5.6742e-02, -1.0794e-01, -1.9884e-02, -2.8014e-02,
          -1.8144e-02, -7.2089e-02, -1.0659e-01, -4.7132e-02, -7.1073e-02,
           2.5898e-01, -6.8692e-02, -5.9335e-02,  1.0095e-01,  2.7785e-03,
          -1.4248e-01, -1.0340e-01, -9.8229e-02,  2.8893e-02, -7.9089e-02,
           1.4736e-01, -2.9456e-03, -5.7306e-02,  1.7827e-01,  1.0001e-01,
           3.2359e-01,  5.4744e-02, -7.5327e-02,  1.1609e-01, -6.7285e-02,
          -2.2617e-02,  1.5121e-01,  1.1619e-01,  4.0007e-02, -1.1279e-01,
          -2.8286e-02, -1.8952e-01,  1.0360e-01,  5.2665e-02, -5.0287e-02,
          -6.0863e-02, -7.5621e-02, -1.7445e-01,  2.9395e-02,  6.6238e-02,
          -8.2340e-03,  3.2121e-02,  8.7107e-02,  3.7411e-02, -7.2104e-02,
          -1.4122e-01,  1.5281e-01,  3.4332e-02,  1.1002e-01,  5.5575e-02,
          -7.9585e-02, -4.7398e-02,  1.5375e-01,  2.4957e-02,  2.0601e-02,
           1.3080e-01, -2.6788e-02, -2.4200e-01,  8.6411e-02,  1.1811e-02],
         [ 4.2684e-02, -2.7031e-03,  1.4298e-02, -1.0974e-01,  1.3387e-01,
          -4.4143e-02, -5.9928e-02, -3.4873e-02,  5.4613e-02, -1.3968e-01,
          -1.0887e-01, -5.5022e-02,  1.9583e-02,  1.2343e-01, -9.9416e-02,
           7.9870e-02,  3.1370e-02, -1.6668e-02, -4.7148e-02,  4.7422e-02,
          -1.4521e-01, -7.0527e-02,  3.6346e-02, -9.6922e-03, -6.4716e-03,
           1.0765e-01,  2.4050e-02, -6.9346e-02,  4.4194e-03, -1.3617e-01,
          -1.4726e-01,  7.6511e-03, -1.4076e-01,  1.6348e-01,  5.5327e-02,
          -1.0936e-01, -4.5788e-04, -8.4764e-02,  2.0963e-01, -1.4858e-01,
           1.3468e-01, -1.6554e-01,  1.9735e-01,  6.8844e-02,  4.0271e-02,
          -1.5831e-01,  1.6706e-01, -3.4029e-02, -8.1691e-02, -3.5286e-02,
          -2.1685e-02,  5.3153e-02, -3.5246e-02,  1.7635e-01, -5.7775e-02,
           1.4218e-01, -3.1825e-02, -3.3248e-03, -1.3808e-02,  2.4296e-01,
          -6.7138e-02, -8.7007e-02, -2.8862e-02, -7.4122e-02,  1.1170e-01,
          -1.2967e-01,  2.0547e-01, -8.3149e-02,  9.3661e-02,  1.1447e-01],
         [ 2.0050e-01, -2.6334e-02,  2.3495e-02,  1.8985e-01,  8.1286e-02,
          -1.8525e-02, -1.4701e-02,  6.1575e-02, -2.6881e-02, -9.7923e-02,
          -3.9525e-02,  1.3943e-02, -2.2108e-01,  8.2307e-02,  5.2262e-02,
          -1.4908e-01,  4.1394e-02,  3.9383e-02, -2.9643e-02,  8.2982e-02,
           1.0812e-01, -8.5415e-02,  1.0080e-01,  1.8355e-02,  9.5582e-02,
           1.0853e-01, -9.6317e-02,  2.3256e-02, -7.9257e-02,  1.3188e-02,
           2.5164e-02, -1.6553e-02,  1.3776e-02, -4.4792e-02, -1.6034e-02,
          -1.7137e-02, -1.6599e-01, -1.1741e-01,  9.7626e-02, -2.0726e-01,
          -6.7859e-02,  9.3904e-02,  5.1553e-02, -1.2293e-02, -7.0319e-02,
          -1.0582e-01,  1.4354e-01, -1.1074e-02, -1.6053e-01,  5.4912e-02,
          -9.8129e-02,  1.7655e-01, -3.9173e-02,  4.7816e-02, -7.5706e-02,
          -3.0862e-02, -3.4565e-02,  1.6844e-01, -2.2667e-01,  7.1861e-02,
          -1.1307e-01, -2.6534e-02,  3.3843e-02, -3.0914e-02,  6.7711e-02,
          -8.4125e-02, -8.5001e-02,  2.6888e-02,  2.5008e-01, -2.2955e-03],
         [ 5.1989e-02, -6.1134e-02, -3.0335e-02,  1.1322e-01, -1.1330e-01,
           3.5291e-02, -2.1876e-02,  6.4551e-03,  4.5225e-02, -1.5405e-01,
           1.2379e-01, -8.2270e-02, -3.0484e-01,  1.0819e-01,  1.3357e-01,
           3.5685e-02, -5.5423e-02,  2.7528e-02, -1.6097e-01,  4.1578e-02,
          -4.5584e-02,  3.7888e-02,  1.2584e-01,  6.4713e-02, -1.0298e-01,
           1.4781e-01, -1.1396e-02,  4.1024e-02, -7.4025e-03,  1.2394e-01,
           4.1668e-02,  4.5705e-03,  1.1440e-01,  6.4562e-02,  1.5776e-01,
           9.3784e-02,  5.3231e-02, -1.4836e-01,  1.1751e-01, -1.3569e-02,
          -2.8206e-01, -5.0432e-02,  1.3207e-01,  9.2000e-02,  1.1093e-01,
           1.1821e-02, -4.3048e-02, -1.0227e-01, -2.2217e-02,  3.5602e-03,
          -2.0588e-02, -4.1376e-02, -3.5228e-02,  1.0398e-01, -2.0375e-01,
          -1.2932e-01,  2.7992e-01,  1.8510e-01, -9.4242e-02, -9.9311e-02,
          -1.2898e-01,  1.2948e-01,  1.0802e-02,  1.8211e-02,  2.0469e-02,
          -4.0870e-02,  1.1680e-02,  1.0431e-01,  4.0065e-02,  3.8654e-02],
         [ 9.9353e-02,  6.4253e-02, -3.4160e-02,  2.7182e-02,  4.1429e-02,
           1.7252e-01, -3.7846e-03,  3.3930e-02, -2.6656e-03, -2.2649e-05,
          -1.1989e-01,  5.7061e-02, -3.8611e-02,  8.7217e-03,  1.0726e-01,
           8.8789e-03,  1.8363e-01,  2.9153e-01, -3.0007e-01,  5.6727e-02,
          -9.9625e-02, -2.1588e-02, -4.1067e-01,  9.8972e-02, -7.5088e-02,
           1.2729e-01,  1.0314e-01, -1.8490e-01,  2.5720e-02,  1.7584e-02,
           5.6193e-03, -2.8768e-02,  3.2220e-02, -1.6139e-01, -1.4008e-02,
          -2.0644e-01, -1.4936e-01, -7.6882e-02,  2.6214e-01,  1.0562e-01,
           2.4997e-02, -9.4328e-03, -8.5398e-02, -5.7442e-02, -8.4854e-02,
           1.2768e-02,  1.0603e-01,  6.0193e-02, -1.3217e-01, -2.2714e-02,
          -2.0082e-01,  2.8143e-02, -1.1461e-01,  1.3467e-01, -1.6298e-01,
          -1.3805e-01, -2.7936e-01,  2.1600e-01,  7.9968e-02, -1.3919e-01,
          -6.6155e-02, -4.8140e-02, -4.2059e-02, -4.0622e-02,  1.4554e-01,
           4.0983e-02,  1.1232e-01,  1.5791e-01, -2.0659e-01,  1.0685e-01],
         [-7.4555e-02, -9.8336e-04,  4.8372e-02, -8.8675e-02,  2.1662e-01,
          -2.5914e-01,  1.1040e-01, -1.7399e-01, -1.1990e-01, -7.3328e-02,
           1.0765e-03,  9.5212e-02,  6.4974e-02,  1.8605e-02, -1.4860e-01,
          -4.2923e-02, -6.6076e-02,  1.3738e-01, -1.2874e-01,  2.2139e-01,
          -1.1609e-01,  2.0101e-01,  1.3427e-01,  1.3271e-01, -7.0087e-02,
          -1.6974e-01, -2.0811e-01, -1.3262e-02, -2.7487e-01, -7.3588e-04,
          -5.0719e-02, -1.2575e-01, -3.1040e-01, -3.2686e-02, -4.4878e-02,
          -1.3627e-01,  4.3030e-02,  3.7402e-02, -9.9832e-02, -2.3791e-01,
           1.1643e-01,  5.1211e-02,  2.7892e-01, -6.7376e-02, -3.3468e-02,
           6.5023e-03, -2.6187e-02,  2.3592e-02, -1.1229e-01,  1.8710e-02,
           1.1554e-01,  4.7669e-02,  1.2471e-01,  1.5595e-02, -8.9394e-02,
           1.0160e-01, -7.0636e-02, -3.6355e-02, -7.2106e-04, -3.2922e-03,
          -3.1203e-01, -8.2269e-02,  1.3747e-01,  2.8943e-01, -2.5065e-01,
          -1.3635e-01, -1.4406e-01, -1.5565e-01, -3.8379e-02, -4.2467e-03],
         [-1.6054e-01, -1.7197e-02, -3.2800e-02, -2.3965e-02, -5.3158e-02,
           1.2333e-02,  8.9618e-03,  6.8235e-02,  9.7215e-02,  1.1321e-02,
           2.4302e-01, -5.5776e-02, -1.1977e-01, -1.0460e-01,  6.6412e-02,
           7.4097e-02,  6.1163e-02, -1.9765e-01,  1.1550e-01,  3.1729e-02,
           1.2110e-01,  3.6793e-02, -2.8560e-01, -5.0376e-02, -1.5311e-01,
          -6.0325e-02, -8.7698e-02,  4.9468e-02, -4.6984e-02,  2.4592e-01,
           2.5259e-02, -1.2948e-01,  1.6054e-01, -1.2765e-01,  1.5064e-01,
          -3.9389e-02, -6.3692e-02, -1.0571e-01, -6.0839e-02,  5.5169e-02,
           5.2784e-02, -1.4439e-01, -2.2454e-02,  6.5641e-02, -1.1294e-02,
           1.6320e-01,  1.6501e-01, -6.6609e-02, -1.6454e-02,  2.5773e-01,
          -1.3573e-01, -9.2846e-02,  5.8761e-03, -8.7776e-02,  3.1023e-02,
          -1.4960e-01, -6.0161e-02, -1.1040e-01,  1.0346e-01, -1.1536e-01,
           1.3904e-01,  1.1988e-01,  1.0675e-01, -6.4604e-02, -1.3922e-01,
           3.0046e-02,  1.1107e-01,  9.9815e-02,  3.8590e-03, -1.1489e-01],
         [ 1.7796e-01,  3.3173e-03,  3.6310e-02, -6.8334e-02, -4.5901e-02,
           1.2242e-01, -5.9630e-02,  6.2765e-02, -1.2207e-01, -1.6675e-01,
           1.0564e-01,  7.8418e-02,  1.1342e-02,  1.2741e-01, -1.5002e-01,
          -1.4442e-01, -5.3747e-02,  5.2537e-02, -6.8010e-02,  2.7846e-02,
          -5.4334e-02, -2.1628e-02,  1.5251e-01, -1.8077e-01,  5.4305e-03,
           8.1400e-02,  2.3313e-01, -1.7239e-01, -2.1218e-01,  9.9754e-02,
          -1.9443e-01,  2.4601e-01,  5.9472e-02, -1.9648e-01, -3.1408e-03,
          -5.7469e-02,  1.4491e-02, -1.4890e-01,  1.9786e-02, -8.8356e-02,
          -6.7325e-02, -5.4330e-02,  6.3219e-02,  7.0149e-05,  2.6016e-02,
           5.6930e-03,  7.0553e-02, -1.0225e-01, -8.5517e-02,  4.9350e-02,
          -1.8276e-01,  2.2137e-02, -6.4787e-02,  1.1811e-01, -1.3203e-01,
          -1.5250e-01,  3.3393e-02,  1.8681e-01, -2.0384e-01, -8.0982e-03,
          -1.6721e-01,  7.6560e-02,  2.3648e-03, -7.9858e-02,  1.3694e-01,
           8.9148e-02, -9.7790e-02, -1.0912e-01, -1.2010e-01, -3.2526e-01],
         [ 1.0796e-02,  1.7112e-01, -8.0794e-02, -1.0201e-02, -1.6678e-01,
           4.8956e-02, -3.3468e-02, -1.1329e-01, -1.4307e-01,  3.8808e-03,
          -1.7007e-01,  1.1081e-01,  3.9384e-02, -8.6496e-02,  1.6035e-02,
           1.3158e-02,  2.2107e-01, -4.7775e-02, -1.9427e-01,  2.0944e-01,
          -7.5677e-02, -1.1798e-01,  1.1025e-01, -1.8711e-01,  3.3253e-02,
          -1.1324e-01,  1.2248e-01, -7.5973e-02,  8.3296e-03, -1.4975e-01,
          -1.0782e-01, -5.1636e-02, -1.3034e-01, -1.0203e-01, -5.1191e-02,
          -7.6037e-02,  3.1529e-02, -1.3401e-01, -7.4482e-02,  2.1677e-01,
           8.4192e-02,  1.0355e-01,  1.0202e-02, -1.0927e-01, -2.2705e-01,
           3.8898e-02,  8.1357e-02,  1.3390e-01,  1.7494e-01, -1.6710e-01,
           9.5442e-02,  8.4279e-02, -2.0884e-01,  2.8360e-01, -2.1652e-02,
           3.0337e-01, -5.0468e-02, -6.0181e-02, -9.9337e-02,  1.5276e-01,
           7.6341e-02,  1.4394e-01,  1.1333e-01,  1.3480e-01,  5.9742e-02,
          -2.8182e-02,  5.9056e-02, -1.5152e-01,  2.1141e-01,  2.2443e-01],
         [ 3.1242e-01, -1.3328e-02, -6.4702e-02, -7.3662e-02,  3.3083e-02,
          -1.8229e-01,  3.9831e-02,  2.8169e-02,  2.9558e-01, -8.1146e-02,
          -5.9358e-02,  1.4800e-01,  6.7160e-02,  5.6184e-02,  6.5894e-02,
           8.5739e-02, -4.2400e-02, -5.4244e-02, -9.8622e-02, -2.8258e-03,
           7.4035e-02, -1.1368e-01, -5.8690e-03, -4.2679e-02,  4.3065e-03,
          -9.3517e-02,  2.3469e-01,  1.0425e-01, -3.9116e-02, -1.1520e-01,
           1.9604e-01,  9.0298e-03, -1.4587e-01, -8.2396e-02,  1.6304e-01,
           9.8176e-02, -1.2156e-01, -1.3441e-01,  1.1860e-01,  1.1502e-01,
          -7.3041e-02,  1.3788e-01, -7.3918e-03,  3.0716e-02,  1.1952e-01,
           1.6556e-01,  4.3813e-02, -1.2945e-01,  1.7765e-01, -1.0628e-01,
           2.0266e-01, -1.4229e-01, -8.0150e-02, -3.0512e-01,  1.6196e-01,
           3.0840e-02,  1.1530e-01, -2.5387e-02, -8.1937e-02,  1.6545e-01,
          -1.4643e-01, -6.8823e-02,  5.6442e-02, -2.6221e-01,  6.0834e-02,
           4.5801e-02,  1.0073e-01, -4.4227e-02,  1.2012e-01, -3.6080e-03],
         [ 2.0685e-01, -1.2266e-02,  3.7424e-02,  1.3822e-01, -7.3270e-02,
           1.1441e-01, -1.2382e-01,  1.2796e-01, -6.4230e-02, -1.5086e-01,
          -1.6847e-02, -1.9173e-01, -6.3245e-02, -3.6363e-02, -3.2275e-02,
           3.1215e-02, -3.0032e-02,  5.2593e-02, -7.6114e-02, -1.1319e-01,
          -4.5392e-02, -1.1279e-03, -1.5751e-01,  1.4057e-02, -1.2208e-02,
           1.8330e-01,  5.4593e-02, -5.1254e-02,  1.5672e-02, -2.0811e-02,
          -4.6908e-02,  1.1040e-01,  3.2581e-03, -6.4124e-02,  9.6250e-02,
          -5.3775e-02,  1.1197e-01, -3.8666e-02, -2.0764e-01,  2.2548e-04,
           2.7937e-02, -1.0908e-01,  3.4961e-01,  1.8435e-02, -6.5997e-02,
           1.2136e-01, -3.5810e-03, -7.4943e-02,  9.9110e-02, -8.0469e-02,
           1.3041e-01, -1.7082e-01,  3.6026e-03, -9.0824e-04,  6.7820e-02,
          -2.5519e-02,  8.1379e-02,  8.5598e-02,  7.6895e-02, -3.6114e-03,
          -5.1131e-02, -7.5736e-03,  1.7272e-02,  6.1835e-02,  1.7413e-01,
          -1.4059e-01,  4.3275e-02,  2.5384e-02,  1.5131e-01,  1.9625e-01],
         [ 1.6178e-01, -7.0462e-02,  5.1837e-02,  1.2976e-01, -1.9617e-01,
          -3.4532e-02,  1.9907e-01,  5.2852e-02, -1.7697e-01,  4.6298e-02,
           2.2926e-01,  8.3025e-02,  1.1260e-01, -1.0937e-01, -5.7699e-02,
           2.3633e-01,  1.5617e-02,  3.5624e-02,  5.5827e-03,  1.3430e-01,
          -2.5599e-03,  2.0441e-01, -5.0584e-02,  1.3386e-01,  1.3817e-01,
           9.8355e-02,  9.8374e-02, -6.2096e-02,  1.0149e-01,  6.8557e-02,
          -2.1888e-01, -1.3535e-01,  1.3701e-01, -1.5107e-01,  1.8643e-01,
          -4.9615e-02,  6.9683e-02,  1.4800e-01, -7.8982e-02, -4.5765e-02,
           5.8402e-02, -6.0298e-02,  4.4689e-02,  2.0854e-02,  5.9750e-02,
           1.0959e-01,  8.6809e-02,  6.2491e-03, -1.4493e-01, -1.5228e-02,
           1.5438e-01, -2.5044e-02, -8.9997e-02,  9.6676e-02,  4.4962e-02,
          -1.1919e-01, -3.1666e-02,  5.8698e-02, -1.8586e-01, -8.2214e-02,
          -2.0919e-02,  6.4359e-02,  1.7290e-02, -1.3625e-01, -9.5965e-03,
          -2.5466e-02,  1.3502e-01,  7.1203e-02, -1.0597e-01, -1.3823e-02],
         [-7.0465e-02, -1.7589e-01, -4.7922e-02, -6.0889e-02,  1.8730e-01,
           4.7207e-02,  1.4650e-01, -1.1668e-01,  6.5600e-02, -2.0100e-02,
           1.7324e-02, -7.2771e-02, -9.2855e-03,  1.0280e-01, -9.2456e-02,
           6.6540e-02, -8.4615e-02, -4.2603e-02,  1.4238e-01,  1.6706e-01,
           1.4694e-01,  7.8026e-02, -2.4749e-02, -6.1624e-02, -3.3867e-01,
          -9.2765e-02, -6.2037e-02, -7.0383e-02,  8.2647e-02, -1.4202e-01,
          -1.3576e-03, -9.3392e-02, -1.0022e-01, -3.3840e-03,  1.5150e-01,
           6.9218e-02,  6.3082e-02,  9.3894e-02, -5.2440e-02,  1.7739e-01,
          -2.6205e-02,  3.1875e-02, -6.3834e-03, -2.5625e-03,  9.3113e-02,
          -1.1761e-01, -1.3483e-01, -8.2253e-02, -2.5895e-01, -2.4529e-05,
          -5.8787e-02, -8.7753e-02, -1.0200e-01, -4.3242e-02,  9.7991e-02,
           1.5287e-01, -7.8606e-02, -7.7933e-02,  3.2688e-02, -9.1445e-02,
           4.1327e-02, -3.6453e-02, -1.6838e-01,  1.1062e-01, -2.5284e-01,
           2.2474e-01,  9.0995e-02, -1.5933e-02,  1.2434e-01, -1.0364e-01]]],
       grad_fn=<StackBackward>), tensor([[[ 9.3221e-02, -2.1853e-02,  2.0682e-02, -1.0197e-02, -3.3932e-01,
          -3.6853e-02, -6.3858e-02,  1.7994e-01,  3.5356e-02,  1.4251e-01,
          -1.2724e-01, -2.8851e-01,  6.0729e-01, -2.4753e-01, -2.5470e-03,
          -2.8043e-01,  2.6954e-01, -8.9303e-03, -1.7162e-01, -2.4749e-01,
           1.8041e-01,  9.5187e-03, -1.3322e-01, -4.4089e-01, -2.6173e-02,
          -1.4541e-01,  4.8343e-01, -1.2250e-01, -7.0039e-01,  2.7829e-01,
          -2.8358e-01,  5.2311e-01,  9.7638e-02, -4.1734e-01,  4.5855e-01,
          -5.2227e-01, -1.9026e-02, -2.6657e-01,  4.6111e-01,  1.0345e-01,
           5.0559e-02, -2.1316e-02,  1.7405e-02, -7.2984e-01, -1.3501e-01,
          -3.7399e-02,  2.6171e-01,  2.6053e-01, -1.7309e-01,  2.8220e-01,
          -3.0814e-03,  4.0326e-01,  8.3211e-02, -3.8659e-03, -2.5465e-01,
          -2.5816e-01, -6.2761e-02,  3.0474e-01, -1.8982e-02, -3.7219e-03,
           6.9788e-02,  9.0356e-02, -7.1155e-02, -1.5040e-01,  2.9454e-01,
          -2.4240e-01, -2.0283e-01,  7.1444e-03, -1.0312e-01, -2.0082e-01],
         [ 1.9737e-01, -5.3190e-02, -3.2687e-01, -3.8976e-02, -1.6248e-01,
          -7.7043e-03, -2.3609e-01, -2.6309e-01, -3.6695e-02, -5.3819e-02,
          -5.2302e-02, -1.3966e-01, -2.4161e-01, -8.2162e-02, -9.0544e-02,
           4.1526e-01, -1.5468e-01, -2.5021e-01,  2.8686e-01,  4.8582e-03,
          -3.1134e-01, -1.5807e-01, -1.7146e-01,  5.1557e-02, -1.4408e-01,
           2.2354e-01, -9.9359e-03, -1.1050e-01,  2.8939e-01,  2.2009e-01,
           5.2792e-01,  8.7768e-02, -1.3312e-01,  2.1684e-01, -9.8586e-02,
          -3.4231e-02,  3.2627e-01,  1.6160e-01,  5.2249e-02, -2.3540e-01,
          -4.0262e-02, -3.0841e-01,  1.8294e-01,  1.2769e-01, -9.5019e-02,
          -1.3013e-01, -1.6222e-01, -3.0249e-01,  6.1805e-02,  1.3391e-01,
          -2.3828e-02,  4.5190e-02,  1.3887e-01,  6.4567e-02, -2.8719e-01,
          -2.5011e-01,  3.0145e-01,  9.5538e-02,  2.4999e-01,  8.9248e-02,
          -2.3376e-01, -8.8702e-02,  2.7605e-01,  9.6764e-02,  6.2590e-02,
           3.2011e-01, -1.2883e-01, -5.0521e-01,  2.5699e-01,  2.0972e-02],
         [ 6.8720e-02, -6.8581e-03,  2.9709e-02, -1.9511e-01,  2.7586e-01,
          -6.7712e-02, -1.2085e-01, -9.8914e-02,  9.1453e-02, -3.3462e-01,
          -2.8225e-01, -1.0812e-01,  3.3569e-02,  2.0245e-01, -2.3233e-01,
           1.6680e-01,  6.0841e-02, -5.7103e-02, -1.0994e-01,  7.3374e-02,
          -2.4406e-01, -1.2747e-01,  9.4365e-02, -1.7129e-02, -1.2252e-02,
           1.7178e-01,  6.7012e-02, -1.3271e-01,  6.4197e-03, -3.2225e-01,
          -3.5327e-01,  1.5679e-02, -3.1131e-01,  2.5785e-01,  9.9131e-02,
          -1.6311e-01, -2.5167e-03, -2.1125e-01,  3.3475e-01, -2.5663e-01,
           2.2486e-01, -3.6455e-01,  3.9049e-01,  3.6275e-01,  8.5612e-02,
          -3.1209e-01,  3.7172e-01, -8.2645e-02, -1.3154e-01, -8.0890e-02,
          -3.6997e-02,  7.5027e-02, -7.6451e-02,  4.9201e-01, -1.7825e-01,
           1.9022e-01, -7.3472e-02, -8.0753e-03, -3.4460e-02,  3.9550e-01,
          -1.4864e-01, -1.5982e-01, -7.5373e-02, -2.5247e-01,  2.3878e-01,
          -2.8945e-01,  4.0671e-01, -1.7517e-01,  1.8692e-01,  1.7226e-01],
         [ 3.3914e-01, -1.0150e-01,  3.6324e-02,  4.9763e-01,  1.7872e-01,
          -5.1438e-02, -3.5377e-02,  1.3797e-01, -5.7502e-02, -1.8878e-01,
          -6.4501e-02,  4.6335e-02, -4.2328e-01,  2.4676e-01,  1.4955e-01,
          -3.1013e-01,  1.2169e-01,  8.5453e-02, -7.8571e-02,  1.8967e-01,
           1.8472e-01, -1.5481e-01,  1.5978e-01,  5.5273e-02,  1.6735e-01,
           2.2305e-01, -2.0697e-01,  6.8470e-02, -1.4087e-01,  2.4627e-02,
           5.1127e-02, -4.1141e-02,  3.4504e-02, -7.2435e-02, -6.3073e-02,
          -3.8727e-02, -2.5438e-01, -2.1037e-01,  1.5405e-01, -6.0112e-01,
          -1.2181e-01,  2.6668e-01,  1.2170e-01, -2.6189e-02, -1.4070e-01,
          -2.3590e-01,  2.6951e-01, -1.9096e-02, -2.0943e-01,  1.3602e-01,
          -1.8962e-01,  3.2909e-01, -5.7176e-02,  1.4603e-01, -1.8357e-01,
          -4.7481e-02, -1.2835e-01,  2.3161e-01, -3.7570e-01,  2.4671e-01,
          -1.6740e-01, -4.9014e-02,  5.9217e-02, -4.8636e-02,  2.2625e-01,
          -1.7269e-01, -1.4470e-01,  5.5398e-02,  5.1291e-01, -3.0618e-03],
         [ 1.0038e-01, -3.5295e-01, -1.1223e-01,  2.0749e-01, -2.1230e-01,
           5.6611e-02, -5.3001e-02,  1.3767e-02,  6.1914e-02, -2.5353e-01,
           2.8447e-01, -1.5193e-01, -5.2399e-01,  2.0529e-01,  1.9352e-01,
           7.7056e-02, -1.7808e-01,  6.5855e-02, -5.0983e-01,  1.5374e-01,
          -8.9672e-02,  6.6706e-02,  3.2300e-01,  2.3437e-01, -2.1439e-01,
           2.0593e-01, -3.0818e-02,  7.6749e-02, -1.2438e-02,  2.4969e-01,
           1.1898e-01,  9.7242e-03,  2.0512e-01,  1.6304e-01,  3.2368e-01,
           2.6813e-01,  8.4527e-02, -2.7434e-01,  1.8818e-01, -2.6560e-02,
          -4.5608e-01, -1.6532e-01,  1.8110e-01,  3.8471e-01,  2.1171e-01,
           3.0737e-02, -7.2112e-02, -2.4098e-01, -3.2850e-02,  1.0286e-02,
          -3.0825e-02, -1.1385e-01, -8.3972e-02,  2.0975e-01, -2.8323e-01,
          -2.6325e-01,  4.1334e-01,  2.4742e-01, -3.1405e-01, -1.7794e-01,
          -3.6114e-01,  2.2747e-01,  6.0010e-02,  7.1858e-02,  4.6038e-02,
          -7.6278e-02,  2.6129e-02,  5.1331e-01,  1.2563e-01,  5.0023e-02],
         [ 2.8786e-01,  1.2624e-01, -6.4549e-02,  4.2915e-02,  1.1021e-01,
           3.3923e-01, -7.4803e-03,  7.1931e-02, -5.6782e-03, -3.4285e-05,
          -1.9512e-01,  1.1535e-01, -5.6018e-02,  1.9738e-02,  2.5366e-01,
           2.7576e-02,  2.9873e-01,  4.3182e-01, -4.7014e-01,  1.6632e-01,
          -2.2095e-01, -5.9088e-02, -6.8763e-01,  2.5503e-01, -1.9865e-01,
           3.4475e-01,  3.3823e-01, -3.9312e-01,  6.3811e-02,  2.6375e-02,
           7.2566e-03, -8.2385e-02,  9.0767e-02, -3.4980e-01, -2.8820e-02,
          -3.7580e-01, -3.5933e-01, -1.0852e-01,  4.5490e-01,  4.0452e-01,
           4.1782e-02, -1.8085e-02, -2.1346e-01, -1.9299e-01, -1.7648e-01,
           2.9896e-02,  1.8513e-01,  1.6495e-01, -1.8540e-01, -3.3192e-02,
          -3.5735e-01,  5.2912e-02, -2.6158e-01,  1.8930e-01, -2.9983e-01,
          -2.6230e-01, -4.3626e-01,  3.6495e-01,  1.4490e-01, -3.1439e-01,
          -1.7082e-01, -1.7548e-01, -1.0599e-01, -9.1801e-02,  2.2451e-01,
           5.7046e-02,  2.5704e-01,  2.2243e-01, -2.9556e-01,  1.6933e-01],
         [-1.2264e-01, -1.6139e-03,  8.3547e-02, -3.4679e-01,  5.1697e-01,
          -5.1913e-01,  2.0079e-01, -2.4473e-01, -3.1554e-01, -1.1166e-01,
           1.5918e-03,  2.5647e-01,  1.1749e-01,  3.2802e-02, -4.7335e-01,
          -6.1141e-02, -1.9030e-01,  2.6436e-01, -2.0826e-01,  3.6943e-01,
          -1.8277e-01,  3.1914e-01,  2.4050e-01,  2.2259e-01, -2.2371e-01,
          -2.5980e-01, -2.9088e-01, -3.2148e-02, -4.3540e-01, -2.1233e-03,
          -2.5849e-01, -2.0965e-01, -6.9691e-01, -4.4344e-02, -1.4205e-01,
          -2.1590e-01,  1.0110e-01,  5.6536e-02, -2.2487e-01, -3.3100e-01,
           1.5426e-01,  1.4187e-01,  4.4135e-01, -1.2731e-01, -9.4499e-02,
           1.5540e-02, -5.7828e-02,  5.8568e-02, -3.6601e-01,  4.0273e-02,
           2.1470e-01,  8.4530e-02,  2.4285e-01,  4.0926e-02, -1.4313e-01,
           1.6460e-01, -4.9634e-01, -7.1330e-02, -1.8752e-03, -4.9345e-03,
          -4.2916e-01, -1.8744e-01,  2.9468e-01,  5.0795e-01, -4.8845e-01,
          -2.6404e-01, -2.8621e-01, -4.5806e-01, -7.6670e-02, -7.2561e-03],
         [-3.1185e-01, -3.9730e-02, -7.0211e-02, -3.5770e-02, -8.4600e-02,
           2.2387e-02,  2.1874e-02,  1.6991e-01,  1.9120e-01,  2.0670e-02,
           4.8225e-01, -1.4302e-01, -2.8126e-01, -3.9548e-01,  1.9028e-01,
           1.5077e-01,  1.4639e-01, -3.5098e-01,  2.5144e-01,  6.4124e-02,
           4.4133e-01,  7.7207e-02, -4.4081e-01, -1.0494e-01, -3.0790e-01,
          -1.6653e-01, -2.5453e-01,  1.2043e-01, -1.3330e-01,  3.0029e-01,
           4.4682e-02, -2.9683e-01,  4.1949e-01, -3.5271e-01,  2.8108e-01,
          -9.5780e-02, -1.0431e-01, -1.6151e-01, -9.9230e-02,  3.2122e-01,
           1.2760e-01, -2.9454e-01, -4.3228e-02,  1.5055e-01, -3.6183e-02,
           1.9809e-01,  2.5835e-01, -9.3577e-02, -2.6526e-02,  4.0571e-01,
          -2.4504e-01, -2.6494e-01,  1.9156e-02, -1.7328e-01,  9.6623e-02,
          -3.4917e-01, -1.3453e-01, -1.9576e-01,  2.3733e-01, -2.5069e-01,
           3.5613e-01,  3.0903e-01,  1.6501e-01, -9.9379e-02, -2.7878e-01,
           4.2593e-02,  2.4134e-01,  2.0081e-01,  5.6253e-03, -2.0653e-01],
         [ 3.0258e-01,  5.6755e-03,  8.0829e-02, -2.3913e-01, -8.2219e-02,
           2.7575e-01, -9.6647e-02,  1.9720e-01, -3.6095e-01, -3.4879e-01,
           1.6223e-01,  2.8609e-01,  3.3609e-02,  2.3220e-01, -2.7162e-01,
          -3.1829e-01, -1.4128e-01,  1.2211e-01, -1.2635e-01,  6.8159e-02,
          -1.0972e-01, -4.7835e-02,  2.3833e-01, -3.9196e-01,  1.6315e-02,
           1.5411e-01,  3.3699e-01, -3.3067e-01, -3.8685e-01,  3.6017e-01,
          -6.0691e-01,  3.2734e-01,  8.0821e-02, -2.8783e-01, -9.6921e-03,
          -1.0512e-01,  2.6337e-02, -2.2179e-01,  4.5302e-02, -1.2896e-01,
          -1.2573e-01, -1.1945e-01,  1.3823e-01,  1.5509e-04,  9.3562e-02,
           9.8471e-03,  1.5868e-01, -2.3289e-01, -1.3579e-01,  1.4495e-01,
          -3.1569e-01,  6.4327e-02, -1.5535e-01,  2.7224e-01, -2.2671e-01,
          -5.3777e-01,  6.8213e-02,  2.7594e-01, -3.5269e-01, -1.9675e-02,
          -2.9704e-01,  1.6667e-01,  3.8123e-03, -1.7539e-01,  3.9340e-01,
           2.8019e-01, -2.4932e-01, -1.9535e-01, -3.2092e-01, -5.1297e-01],
         [ 3.5070e-02,  2.5370e-01, -1.7789e-01, -2.4926e-02, -2.8720e-01,
           1.8292e-01, -6.8666e-02, -1.6878e-01, -2.6161e-01,  4.9258e-03,
          -3.0194e-01,  2.8577e-01,  8.3778e-02, -2.4808e-01,  5.6104e-02,
           3.0661e-02,  2.8402e-01, -8.4572e-02, -4.0580e-01,  3.0525e-01,
          -1.6047e-01, -2.5737e-01,  2.1379e-01, -2.5986e-01,  1.0583e-01,
          -2.0130e-01,  1.9707e-01, -1.3183e-01,  1.5425e-02, -3.3098e-01,
          -2.2770e-01, -1.4879e-01, -3.0058e-01, -1.8496e-01, -9.7611e-02,
          -1.3701e-01,  1.3882e-01, -2.6926e-01, -2.1139e-01,  3.6060e-01,
           2.0052e-01,  2.3758e-01,  4.1201e-02, -1.8915e-01, -5.1263e-01,
           1.8563e-01,  1.3131e-01,  3.6116e-01,  5.2227e-01, -4.8509e-01,
           2.4472e-01,  3.0810e-01, -5.0771e-01,  6.0899e-01, -8.8654e-02,
           5.8784e-01, -1.9630e-01, -1.7372e-01, -1.9244e-01,  2.9555e-01,
           1.0260e-01,  2.6353e-01,  2.2270e-01,  2.0949e-01,  1.4385e-01,
          -5.0362e-02,  1.7004e-01, -2.1097e-01,  3.4458e-01,  3.9557e-01],
         [ 5.3647e-01, -2.9999e-02, -1.4112e-01, -3.3890e-01,  1.2244e-01,
          -3.3227e-01,  8.2622e-02,  4.5105e-02,  3.6365e-01, -2.9853e-01,
          -1.0386e-01,  2.6826e-01,  2.1300e-01,  1.0996e-01,  2.0498e-01,
           2.0925e-01, -1.2533e-01, -1.0446e-01, -2.4071e-01, -8.3336e-03,
           1.2045e-01, -3.0621e-01, -1.3391e-02, -1.1961e-01,  1.2232e-02,
          -1.2337e-01,  4.6494e-01,  2.0077e-01, -6.9105e-02, -1.6854e-01,
           3.4292e-01,  2.5582e-02, -4.6363e-01, -2.4076e-01,  2.6571e-01,
           2.5924e-01, -2.8040e-01, -4.5906e-01,  2.1933e-01,  1.5992e-01,
          -1.9775e-01,  2.7314e-01, -1.2692e-02,  9.7699e-02,  2.2890e-01,
           3.1681e-01,  1.1788e-01, -2.2902e-01,  2.7566e-01, -2.1298e-01,
           3.5494e-01, -2.2925e-01, -1.3976e-01, -5.1721e-01,  4.4678e-01,
           4.9507e-02,  5.1684e-01, -5.3684e-02, -1.8790e-01,  4.4092e-01,
          -4.4377e-01, -1.0593e-01,  9.9731e-02, -5.7961e-01,  1.4841e-01,
           2.3734e-01,  1.6730e-01, -1.7277e-01,  3.8250e-01, -5.3172e-03],
         [ 3.4999e-01, -4.2146e-02,  4.9193e-02,  3.2536e-01, -1.4589e-01,
           1.9458e-01, -2.8026e-01,  2.9363e-01, -1.0611e-01, -2.1341e-01,
          -7.7877e-02, -2.6966e-01, -1.4948e-01, -7.8020e-02, -6.2451e-02,
           8.4386e-02, -3.8987e-02,  1.8255e-01, -1.9776e-01, -1.9268e-01,
          -2.1243e-01, -2.0813e-03, -2.6093e-01,  2.6608e-02, -2.0829e-02,
           4.2661e-01,  1.6975e-01, -7.6406e-02,  2.2732e-02, -6.5018e-02,
          -9.1071e-02,  1.7558e-01,  5.8926e-03, -1.5338e-01,  2.4600e-01,
          -7.3771e-02,  2.2762e-01, -7.4426e-02, -2.6732e-01,  5.8091e-04,
           3.7521e-02, -1.5464e-01,  5.2612e-01,  4.5389e-02, -1.9336e-01,
           2.4077e-01, -5.4095e-03, -1.3343e-01,  2.6770e-01, -1.8485e-01,
           3.1330e-01, -2.6992e-01,  6.5225e-03, -1.7711e-03,  2.2823e-01,
          -5.1986e-02,  1.8584e-01,  1.8162e-01,  1.6720e-01, -6.7403e-03,
          -1.2307e-01, -1.6137e-02,  3.9924e-02,  2.8196e-01,  4.1205e-01,
          -2.6285e-01,  8.5320e-02,  5.5295e-02,  3.4172e-01,  2.9445e-01],
         [ 3.3833e-01, -1.0037e-01,  1.0737e-01,  2.1281e-01, -3.2110e-01,
          -6.0926e-02,  4.3145e-01,  1.2466e-01, -4.1285e-01,  1.5106e-01,
           4.4431e-01,  1.3922e-01,  2.1206e-01, -2.8239e-01, -1.0274e-01,
           4.0948e-01,  3.1457e-02,  8.7206e-02,  1.5461e-02,  2.0519e-01,
          -6.5999e-03,  3.8007e-01, -1.4668e-01,  2.4008e-01,  2.0687e-01,
           2.1578e-01,  2.7252e-01, -1.1205e-01,  2.6117e-01,  3.0541e-01,
          -4.0405e-01, -2.0155e-01,  2.3806e-01, -3.5600e-01,  4.2903e-01,
          -1.1373e-01,  1.3673e-01,  2.2938e-01, -1.3081e-01, -8.3466e-02,
           1.3446e-01, -1.7567e-01,  8.1113e-02,  4.6026e-02,  2.6252e-01,
           1.8118e-01,  1.6280e-01,  1.4171e-02, -2.5638e-01, -2.6181e-02,
           2.4439e-01, -7.4716e-02, -2.0319e-01,  2.0091e-01,  9.5616e-02,
          -3.1175e-01, -4.1305e-02,  1.0913e-01, -2.8723e-01, -1.4918e-01,
          -7.6587e-02,  2.2088e-01,  5.5809e-02, -2.1610e-01, -2.0056e-02,
          -4.7992e-02,  2.9671e-01,  1.8235e-01, -2.8683e-01, -3.0831e-02],
         [-3.1540e-01, -2.5583e-01, -7.3563e-02, -1.1689e-01,  3.3020e-01,
           6.4550e-02,  3.2055e-01, -2.9034e-01,  1.1113e-01, -4.4030e-02,
           3.1428e-02, -1.7948e-01, -2.7546e-02,  1.8931e-01, -3.6649e-01,
           1.1624e-01, -5.6002e-01, -6.4316e-02,  3.0093e-01,  2.4970e-01,
           2.9886e-01,  1.2596e-01, -9.0692e-02, -1.8891e-01, -4.6957e-01,
          -1.9938e-01, -9.2532e-02, -3.2602e-01,  1.9859e-01, -1.8314e-01,
          -3.5364e-03, -2.3460e-01, -1.9716e-01, -9.2793e-03,  2.2748e-01,
           1.3924e-01,  2.6205e-01,  2.0593e-01, -9.5032e-02,  2.7690e-01,
          -1.0782e-01,  8.8690e-02, -1.1525e-02, -4.1536e-03,  2.6947e-01,
          -2.1392e-01, -2.1780e-01, -1.2589e-01, -4.8847e-01, -3.9643e-05,
          -1.2109e-01, -2.8154e-01, -2.5572e-01, -1.3344e-01,  1.3336e-01,
           3.6146e-01, -4.5415e-01, -1.8555e-01,  6.9672e-02, -2.3003e-01,
           9.7687e-02, -6.9592e-02, -2.1537e-01,  1.8036e-01, -3.5553e-01,
           4.0229e-01,  1.4811e-01, -4.8389e-02,  3.0865e-01, -2.4413e-01]]],
       grad_fn=<StackBackward>)))
>>>

Would you mind monitoring the system status with tegrastats on the other console at the same time?
If the issue is caused by out-of-memory, please try to add some swap memory to see if it helps.

Thanks.

Yep that fixed it, thanks very much!