@Usama_Hasan
Error:
builtins.RuntimeError: size mismatch, m1: [512 x 517], m2: [512 x 505] at c:\a\w\1\s\windows\pytorch\aten\src\th\generic/THTensorMath.cpp:940
Input data:
<bound method DataFrame.info of         duration     src_bytes  ...  service_vmnet  service_whois
0       0.000000  3.558064e-07  ...              0              0
1       0.000000  1.057999e-07  ...              0              0
2       0.000000  0.000000e+00  ...              0              0
3       0.000000  1.681203e-07  ...              0              0
4       0.000000  1.442067e-07  ...              0              0
...          ...           ...  ...            ...            ...
125968  0.000000  0.000000e+00  ...              0              0
125969  0.000186  7.608895e-08  ...              0              0
125970  0.000000  1.616709e-06  ...              0              0
125971  0.000000  0.000000e+00  ...              0              0
125972  0.000000  1.094232e-07  ...              0              0

[125973 rows x 117 columns]>
tensor([[0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 7.2466e-10, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [1.2958e-01, 1.0652e-07, 8.0157e-08,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        ...,
        [0.0000e+00, 3.0436e-08, 3.2063e-08,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 0.0000e+00, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00],
        [0.0000e+00, 7.4785e-07, 0.0000e+00,  ..., 0.0000e+00, 0.0000e+00,
         0.0000e+00]])
tensor([[0., 1., 0., 0., 0.],
        [0., 0., 1., 0., 0.],
        [1., 0., 0., 0., 0.],
        ...,
        [1., 0., 0., 0., 0.],
        [0., 1., 0., 0., 0.],
        [0., 1., 0., 0., 0.]])