Thank you so much for your reply! I will do that. Although how do I calculate training accuracy from raw logits? Because the 3 values that will be returned can be either both positive or negative (I guess).