Связь между спецификациями видеокарты и параметрами yolov3 [net] в даркнете - PullRequest
0 голосов
/ 27 мая 2020

Я пытаюсь использовать видеокарту NVIDIA GT 1030 2GB (ядра CUDA: 384) для обучения пользовательского объекта (2K изображений с размером 90x90), используя параметры yolov3 [net]:

batch=64 
subdivisions=16
width=416 
height=416

Но мой результат из-за нехватки памяти (я знаю, потому что, если я попытаюсь использовать крошечный Yolov3, работает) darknet: ./src/cuda.c:36: check_error: Assertion '0' failed.

Какова взаимосвязь между спецификациями видеокарты (количество CUDAS и ГБ памяти) и yolov3 [net] параметры (пакет, подразделения, ширина, высота) в дар knet? Моя цель - размер видеокарты, соответствующей моей задаче, без необходимости покупать дорогую видеокарту.

...