mate, this really helped in making progress.
The best result i’ve gotten by making these changes is a validation loss of 0.62 (BCE) and auc 0.69 which is much higher than what i was getting earlier on validation.