Skip to content

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

Description

@guo560

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions