Hi Ptrblk,
How are you? I hope you are avoiding the worst that covid is throwing at you.
I am using 1.4.0 version.
Changing the loss function from F.binary_cross_entropy_with_logits to torch.nn.BCELoss and putting the sigmoid function at end of the network seems to have resolved the error.
I think there maybe an issue with F.binary_cross_entropy_with_logits and binary_cross_entropy?
I have decreased the learning rate as well by an order of magnitude and the model is now running.
I will post the stack trace tomorrow once the model has finished training.
chaslie