Splitting data in assignment C2W1 in CNN with tf course

this thing takes forever and does not run, what should I do, I tried switching from TPU to GPU and changing settings. still doesn’t wanna run

Does this topic from week 2 help?