Erro

loss_treino, loss_teste= [], []
acc_treino, acc_teste = [], []

dados_teste, rotulos_teste = amostrar(50)

for epoca in range(100):

  dados_tns, rotulos_tns = amostrar(50)
  loss, acc = forward(dados_tns, rotulos_tns, 'treino', epoca)
  loss_treino.append(loss)
  acc_treino.append(acc)

  loss, acc = forward(dados_teste, rotulos_teste, 'teste', epoca)
  loss_teste.append(loss)
  acc_teste.append(acc)

Começou a rodar, fez três épocas e travou

<ipython-input-27-c7c181dc5d90> in name2tensor(name)
      6   name_tns = torch.zeros(len(name), len(dicionario))
      7   for k, letra in enumerate(name.decode("utf-8")):
----> 8     name_tns[k][dicionario.index(letra)] = 1
      9 
     10   return name_tns

Etá dando um erro nessa parte e eu não estou conseguindo arrumar

import string
dicionario = string.ascii_letters +" '-"

#Abreu
def name2tensor(name):
  name_tns = torch.zeros(len(name), len(dicionario))
  for k, letra in enumerate(name.decode("utf-8")):
    name_tns[k][dicionario.index(letra)] = 1

  return name_tns

print(dicionario)
dados[0][10], name2tensor(dados[0][10])