Bom dia.
A iteração na qual o travamento inicia é entre 3 e 5.
Isso demora cerca de 1 até 4 min.
Estou usando o chrome :), devorador de ram.
Estou fazendo um teste limpando os outputs seguidamente, está indo 🙂
Uma pergunta importante, você uso o colab hospedado ou conecta ele localmente?
Pergunto pela questão da memória que você citou.
Att Lúcio