Error On Traffic Dataset

[rajeevbaalwan]

Getting error when executing scripts/run_traffic.sh

  File "/mnt/hdd1/rajeevyadav/bc/D-PAD/singlecard.py", line 123, in <module>
    exp.train(setting)
  File "/mnt/hdd1/rajeevyadav/bc/D-PAD/experiments/exp_ETT.py", line 192, in train
    pred, true = self._process_one_batch_DPAD(batch_x, batch_y)
  File "/mnt/hdd1/rajeevyadav/bc/D-PAD/experiments/exp_ETT.py", line 324, in _process_one_batch_DPAD
    outputs = self.model(batch_x)
  File "/mnt/hdd1/rajeevyadav/anaconda3/envs/dpad/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/mnt/hdd1/rajeevyadav/anaconda3/envs/dpad/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
  File "/mnt/hdd1/rajeevyadav/bc/D-PAD/model/D_PAD_SEBlock.py", line 241, in forward
    x = x * self.affine_weight + self.affine_bias
RuntimeError: The size of tensor a (861) must match the size of tensor b (862) at non-singleton dimension 2```