Я использую модуль обнаружения объектов для классификации изображений.ОС: Ubuntu 18.04 LTS Python: 3.6.7 VirtualEnv: Версия: 16.4.3 Версия Pip3 внутри virtualenv: 19.0.3 Версия TensorFlow: 1.13.1 Версия Protoc: 3.0.0-9
Работа в Windows virtualenv иgoogle-colab.
python3 legacy / train.py --logtostderr --train_dir = training / --pipeline_config_path = training / ssd_mobilenet_v1_pets.config
ИНФОРМАЦИЯ: тензор потока: глобальный шаг 1: потеря =18.5013 (48,934 сек / шаг) ИНФОРМАЦИЯ: тензор потока: Закончено обучение!Сохранение модели на диск./home/priyank/venv/lib/python3.6/site-packages/tensorflow/python/summary/writer/writer.py:386: UserWarning: попытка использовать закрытый FileWriter.Операция будет noop, если FileWriter не будет открыто заново.warnings.warn («Попытка использовать закрытый FileWriter.» Traceback (последний вызов был последним): файл «legacy / train.py», строка 184, в файле tf.app.run () »/ home / priyank / venv /lib / python3.6 / site-packages / tenorflow / python / platform / app.py ", строка 125, в файле _sys.exit (main (argv)) в файле run" /home/priyank/venv/lib/python3.6/site-packages / tenorflow / python / util / deprecation.py ", строка 324, в new_func return func (* args, ** kwargs) Файл" legacy / train.py ", строка 180, в основном файле graph_hook_fn = graph_rewriter_fn)"/home/priyank/venv/models-master/research/object_detection/legject/trainer.py ", строка 416, в заставке поезда = заставка) Файл" /home/priyank/venv/lib/python3.6/site-packages/tenorsflow / contrib / slim / python / slim / learning.py ", строка 785, в поезде ignore_live_threads = ignore_live_threads) Файл" /home/priyank/venv/lib/python3.6/site-packages/tensorflow/python/training/supervisor.py ", строка 832, в остановке файл ignore_live_threads = ignore_live_threads)" /home/priyank/venv/lib/python3.6/site-packages/tensorflow/python/training /ordinator.py ", строка 389, в файле join.reraise (* self._exc_info_to_raise), файл" /home/priyank/venv/lib/python3.6/site-packages/six.py ", строка 693, в ререйзеподнять значение файла "/home/priyank/venv/lib/python3.6/site-packages/tensorflow/python/training/queue_runner_impl.py", строка 257, в файле _run enqueue_callable () "/ home / priyank / venv / lib/python3.6/site-packages/tensorflow/python/client/session.py ", строка 1257, в _single_operation_run self._call_tf_sessionrun (файл отсутствует, {}, [], target_list, отсутствует)" / home / priyank / venv /lib / python3.6 / site-packages / tenorflow / python / client / session.py ", строка 1407, в _call_tf_sessionrun run_metadata) ensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM при выделении тензора с формой [15,1,1755,2777,3] и введите float on / job: localhost / replica: 0 / task: 0 / device: CPU: 0 от процессора-распределителя [[{{node batch}}]]] Подсказка: если вы хотите увидетьсписок выделенных тензоров, когда происходит OOM, добавьте report_tensor_allocations_upon_oom в RunOptions для текущего alloинформация о катионах.