There is one more possible clue: even though it would crash immediately with above error most of time, there are some rare cases it can train for around 2-3 epoch, and then crash with the same error.
tf_noob
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution