How to calculate epoch loss when using BCEWithLogitsLoss

I already replied to him about this topic on a different post: Filter Output Using Mask.

1 Like