Training validation stops after a couple of steps

Same issue here. Any help would be appreciated.