Получение AttributeError в Google Colab - PullRequest
Новая
       51

Получение AttributeError в Google Colab

0 голосов
/ 15 марта 2020

Я обучаю модель обнаружения объектов с использованием библиотеки Image AI в Google Colab.

Я получаю следующую ошибку 

AttributeError: '_TfDeviceCaptureOp' object has no attribute '_set_device_from_string'

Вот как эта ошибка 

Generating anchor boxes for training images and annotation...
Average IOU for 9 anchors: 0.98
Anchor Boxes generated.
Detection configuration saved in  /content/drive/My Drive/ColabNotebooks/GoogleColabnotebooks/Malaria_object_detection/dataset/json/detection_config.json
Training on:    ['infected', 'uninfected']
Training with Batch Size:  4
Number of Experiments:  200
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-14-b2bf9748be75> in <module>()
      4 trainer.setDataDirectory(data_directory=data_path)
      5 trainer.setTrainConfig(object_names_array=["infected","uninfected"], batch_size=4, num_experiments=200)
----> 6 trainer.trainModel()

/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/ops.py in _apply_device_functions(self, op)
   4396       # strings, since identity checks are faster than equality checks.
   4397       if device_string is not prior_device_string:
-> 4398         op._set_device_from_string(device_string)
   4399         prior_device_string = device_string
   4400     op._device_code_locations = self._snapshot_device_function_stack_metadata()

AttributeError: '_TfDeviceCaptureOp' object has no attribute '_set_device_from_string'

Я не получаю ошибку, когда я запускаю тот же код на моем ноутбуке. Ниже мой код 

from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory=data_path)
trainer.setTrainConfig(object_names_array=["infected","uninfected"], batch_size=4, num_experiments=200)
trainer.trainModel()

1 Ответ

0 голосов
/ 17 марта 2020

Можете ли вы проверить, запустив ту же версию tenorflow на вашем ноутбуке и Colab. По умолчанию colab загружает версию tenorflow 1.15.0. 

import tensorflow as tf
print(tf.__version__)

Вывод: 

1.15.0

Вы можете установить требуемую версию tenorfow в colab, используя приведенный ниже пример - 

!pip install tensorflow==2.0.0
tf.__version__
...