search

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.
https://deepctr-torch.readthedocs.io/en/latest/index.html
Apache License 2.0
2.95k stars 696 forks source link

IndexError: too many indices for tensor of dimension 1 #263

Open guo560 opened 1 year ago

guo560 commented 1 year ago

Describe the bug(问题描述) image 在basemodel的fit函数中，若batch_size为1，且model为多任务模型时，会导致y_pred的第0个维度被丢弃，从而产生错误。