IndexError: too many indices for tensor of dimension 2

[ushna27]

Hi,

I am trying to run the train_chart.py with lien data. I am getting the below error-

total parameters: 198592138 start prefetching data... shuffling indices... setting learning rate to: 0.00025 training start... ['read.txt'] 0%| | 0/50000 [00:00<?, ?it/s] Traceback (most recent call last): File "train_chart.py", line 217, in train(training_dbs, validation_db, args.start_iter) File "train_chart.py", line 158, in train training_loss = nnet.train(**training) File "C:\Users\opc.INSTANCE-GPI\Documents\DeepRule-master\nnet\py_factory.py", line 81, in train loss = self.network(xs, ys) File "C:\Users\opc.INSTANCE-GPI\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "C:\Users\opc.INSTANCE-GPI\Documents\DeepRule-master\nnet\py_factory.py", line 20, in forward loss = self.loss(preds, ys, **kwargs) File "C:\Users\opc.INSTANCE-GPI\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "C:\Users\opc.INSTANCE-GPI\Documents\DeepRule-master\models\py_utils\kp.py", line 2230, in forward pull, push = self.ae_loss(key_tag_grouped, gt_mask_grouped) File "C:\Users\opc.INSTANCE-GPI\Documents\DeepRule-master\models\py_utils\kp_utils.py", line 592, in _ae_line_loss tag_full[1-mask_full] = 0 IndexError: too many indices for tensor of dimension 2

Can you provide a solution?