我正在尝试使用 ImageAI 库训练对象检测模型,这是我的代码。
from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory='Datasets/hololens')
trainer.setTrainConfig(object_names_array=['hololens'], batch_size=4, num_experiments=100,
train_from_pretrained_model='pretrained-yolov3.h5')
trainer.trainModel()
然而,训练只是中途停止/暂停/挂起。它会在显示错误“Profiler session tear down”后立即停止,如下所示:
2/480 [.......................] - ETA:4:51 - 损失:183.8092 - yolo_layer_loss:25.4697 - yolo_layer_1_loss: 50.0500 - yolo_layer_2_loss: 96.7067 2021-07-30 14:32:31.192212: I tensorflow/core/profiler/lib/profiler_session.cc:71] Profiler 会话收集数据。2021-07-30 14:32:31.196004:I tensorflow/core/profiler/internal/gpu/cupti_tracer.cc:1487] CUPTI 活动缓冲区已刷新 2021-07-30 14:32:31.449851:I tensorflow/core/profiler/ internal/gpu/cupti_collector.cc:228] GpuTracer 已经收集了 5713 个回调 api 事件和 5681 个活动事件。2021-07-30 14:32:31.558638:我 tensorflow/core/profiler/lib/profiler_session.cc:172] Profiler 会话拆除
我正在使用 GTX 1080Ti、Tensorflow 2.4.1、CUDA v11。我一直在训练图像分类模型,但是当我尝试运行这个对象检测模型训练时,就会出现这个问题。
先感谢您!:)