Thank you for your quick response! I tried what you recommended and trained on only 10 samples and ran 1000 epochs to make sure that I overfitted. Here’s what the model outputted for those 10 samples.
