Выберите задачу из цикла Trax

#python #trax

#python #trax

Вопрос:

У меня есть объект цикла Trax, из которого я хотел бы извлечь объект задачи. Приведенный ниже код возвращает ошибку AttributeError: 'Loop' object has no attribute '_task' . У вас есть предложения о том, как это исправить?

Справочная информация: я пытаюсь запустить на графическом процессоре настройку кода назначения W4 курса Coursera «Процесс естественного языка с моделями внимания«.

 test_loop = training_loop(ReformerLM, train_stream, eval_stream)
type(test_loop)
train_task = test_loop._task
 

Вот сообщение об ошибке:

 <class 'trax.supervised.training.Loop'>

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-34-937f26e690d0> in <module>
      1 test_loop = training_loop(ReformerLM, train_stream, eval_stream)
      2 print(type(test_loop))
----> 3 train_task = test_loop._task
      4 #eval_task = test_loop._eval_task
      5 

AttributeError: 'Loop' object has no attribute '_task'
 

Комментарии:

1. Код отлично работает в записной книжке Coursera … это потому, что они используют TPU?

Ответ №1:

Нет доступа к курсу coursera:/, можете ли вы попробовать следующие свойства?

  1. test_loop.tasks , который возвращает учебные задачи.
  2. test_loop.eval_tasks , который возвращает задачи оценки.

Вы можете найти дополнительные свойства в документе trax: https://trax-ml.readthedocs.io/en/latest/trax.supervised.html#trax.supervised.training.Loop.tasks