It sounds like you’ve got most of the issues correctly handled, but the one thing that is perhaps least well explained is that you need to transpose the labels and logits. Here’s a recent thread about that.
BTW no-one can simply examine your code directly. Only the course staff have that “superpower”. If the mentors can’t help without looking at your code, we can share code using private DM messages.