yeah they don’t seem that imbalanced.
hmm, can you train the net with CrossEntropyLoss or BCEwithLogits and then report the loss? (one epoch will suffice)
cause the code seems right and I don’t see anything wrong with it