При использовании cuda вы можете получить эту общую ошибку, которая не очень полезна.Вместо этого попробуйте переключиться на устройство cpu, device = torch.device("cpu"), чтобы увидеть фактическую трассировку стека ошибок.
В моем случае проблема была вызвана тем, что двоичная кросс-энтропия ожидала, что входные значения будут между 0 ~ 1,но я отправлял значения от -1 до 1. Применение sigmoid к выходу решило эту проблему.