Я работаю над Google Colab с графическим процессором.Я хочу провести некоторое обучение на colab, так как он предоставляет графический процессор, который может быть полезен для сокращения времени обучения.Я загрузил папку, содержащую все файлы на диске Google.Затем я установил google drive с colab, используя этот код:
from google.colab import drive
drive.mount('/content/gdrive')
Когда я запускаю команду для обучения, я получаю одну и ту же ошибку снова и снова.Команда и ошибка отображаются:
!python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_coco.config
Ошибка:
Traceback (most recent call last):
File "train.py", line 49, in <module>
from object_detection.builders import dataset_builder
ModuleNotFoundError: No module named 'object_detection'
Всякий раз, когда проблема такого типа возникала на моей локальной машине, проблема всегда возникает из-занеправильная настройка PYTHONPATH
.
Я уже пытался установить PYTHONPATH
на colab, но одна и та же ошибка возникает снова и снова.Пробные команды для установки PYTHONPATH:
Первая попытка:
!export PYTHONPATH="${PYTHONPATH}:/content/gdrive/'My Drive'/tensorflow1/models:/content/gdrive/'My Drive'/tensorflow1/models/research:/content/gdrive/'My Drive'/tensorflow1/models/research/slim:/content/gdrive/'My Drive'/tensorflow1/models/research/object_detection"
Вторая попытка:
!export PYTHONPATH=$PYTHONPATH:/content/gdrive/'My Drive'/tensorflow1/models:/content/gdrive/'My Drive'/tensorflow1/models/research:/content/gdrive/'My Drive'/tensorflow1/models/research/slim:/content/gdrive/'My Drive'/tensorflow1/models/research/object_detection
Третья попытка:
os.environ['PYTHONPATH'] = "/content/gdrive/'My Drive'/tensorflow1/models:/content/gdrive/'My Drive'/tensorflow1/models/research:/content/gdrive/'My Drive'/tensorflow1/models/research/slim:/content/gdrive/My Drive/tensorflow1/models/research/object_detection"
Четвертая попытка:
%set_env PYTHONPATH=":/content/gdrive/My\ Drive/tensorflow1/models:/content/gdrive/My\ Drive/tensorflow1/models/research:/content/gdrive/My\ Drive/tensorflow1/models/research/slim:/content/gdrive/My\ Drive/tensorflow1/models/research/object_detection"
Ни одна из приведенных выше команд не разрешилаошибка.Итак, мне нужно знать, как правильно установить PYTHONPATH на colab, и если установка PYTHONPATH не является правильным способом преодоления этой ошибки, то что мне следует попробовать?