Traceback (most recent call last):


 "test.py", line 32, in <module>

for i, (data,_) in enumerate(dataset):


 "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 212, in __next__

return self._process_next_batch(batch)


 "/usr/local/lib/python2.7/dist-packages/torch/utils/data/dataloader.py", line 239, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
TypeError: Traceback (most recent call last):

Traceback (most recent call last):


 "train_ic15.py", line 283, in <module>

main(args)


 "train_ic15.py", line 251, in main

train_loss, train_te_acc, train_ke_acc, train_te_iou, train_ke_iou = train(train_loader, model, dice_loss, optimizer, epoch)


 "train_ic15.py", line 112, in train

for batch_idx, (imgs, gt_texts, gt_kernels, training_masks) in enumerate(train_loader):


 "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 336, in __next__

return self._process_next_batch(batch)


 "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 357, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
TypeError: function takes exactly 5 arguments (1 given)
