optimized_execution () принимает 1 позиционный аргумент, но 2 были даны

Question

Я слежу за документами Pytorch Sagemaker здесь и застрял в этой строке

torch.jit.optimized_execution(True, {'target_device': 'eia:device ordinal'})

Когда я запускаю его, я получаю сообщение об ошибке optimized_execution() takes 1 positional argument but 2 were given. Я использую Pytorch 1.3.1, но я пробовал с 1.4.0 и столкнулся с аналогичными проблемами. Могу ли я использовать оптимизированное выполнение без этого второго аргумента? Как мне указать ускоритель?

Answer 1

(для удобства я буду ссылаться на инфраструктуру PyTorch с включенным логическим выводом Elasti c как "PyTorch-EI")

Используете ли вы SageMaker через ноутбук или хостинг? Поддержка ноутбуков SageMaker в настоящее время не выпускается, поэтому нет официальной среды ядра ноутбука / Conda, которую вы можете активировать, которая будет иметь инфраструктуру Pylasch с поддержкой Elasti c Inference. Однако вы можете создать свой собственный, активировав среду pytorch_p36 (которая имеет стандартный PyTorch), удалив PyTorch, а затем установив его с помощью колеса фреймворка PyTorch-EI 1.3.1 - связанного с здесь .

Хостинг SageMaker в настоящее время поддерживает PyTorch-EI из коробки. Если вы в настоящее время используете хостинг SageMaker и у вас есть проблемы, поделитесь некоторыми из своего кода вывода + какой контейнер вы используете. Список контейнеров можно найти по адресу здесь .

Также обратите внимание, что EC2 в настоящее время поддерживает вывод Elasti c через DLAMI v27.0. Имя среды Conda amazonei_pytorch_p36.

Answer 2

Я неправильно прочитал документацию. Это говорит the Elastic Inference enabled PyTorch framework accepts two parameters for this context, while the vanilla PyTorch framework accepts only one parameter.. Это должно работать для torch 1.3.1, потому что это все, что поддерживает sagemaker на данный момент.