Skip to content

Using a target size (torch.Size([64])) that is different to the input size (torch.Size([64, 1]) #9

@astraleosun

Description

@astraleosun

loss = F.binary_cross_entropy(y_pred, y)

Hello,
First I got a ValueError: Using a target size (torch.Size([64])) that is different to the input size (torch.Size([64, 1])) is deprecated. Please ensure they have the same size.
Then I altered y_pred to y_pred.squeeze().detach(), but I got a new error showing element 0 of tensors does not require grad and does not have a grad_fn where in line 93, loss.backward().
So can you help me? what should I do?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions