Exponenta Banner

coder.CuDNNConfig

Параметры, чтобы сконфигурировать генерацию кода глубокого обучения с библиотекой CUDA Deep Neural Network

расширьте все на странице

Описание

coder.CuDNNConfig объект содержит NVIDIA® cuDNN определенные параметры что codegen использование для генерации кода CUDA® для глубоких нейронных сетей.

Использовать coder.CuDNNConfig объект для генерации кода, присвойте его DeepLearningConfig свойство coder.gpuConfig возразите, что вы передаете codegen.

Создание

Синтаксис

deepLearningCfg = coder.DeepLearningConfig('cudnn')

Описание

пример

deepLearningCfg = coder.DeepLearningConfig('cudnn') создает coder.CuDNNConfig объект для генерации кода глубокого обучения при помощи cuDNN библиотеки.

Свойства

развернуть все

Включите или отключите автоматическую настраивающую опцию. Включение автоматической настройки позволяет cuDNN библиотеке находить самые быстрые алгоритмы свертки. Это увеличивает производительность для больших сетей, таких как SegNet и ResNet

Задайте точность входа типа данных тензора к сети. При выполнении вывода в 32-битных плаваниях используйте 'fp32'. Для 8-битного целого числа используйте 'int8'. Значением по умолчанию является 'fp32'.

INT8 точность требует, чтобы графический процессор CUDA с минимумом вычислил возможность 6,1. Вычислите возможность 6,2, не поддерживает INT8 точность. Используйте ComputeCapability свойство GpuConfig объект установить соответствующее вычисляет значение возможности.

Местоположение MAT-файла, содержащего калибровочные данные. Значением по умолчанию является ''. Эта опция применима только когда DataType установлен в 'int8'.

Для получения дополнительной информации смотрите Генерацию кода для Квантованных Нейронных сетей для глубокого обучения.

Значение только для чтения, которое задает имя целевой библиотеки.

Примеры

свернуть все

Создайте функцию точки входа resnet_predict это использует coder.loadDeepLearningNetwork функционируйте, чтобы загрузить resnet50 SeriesNetwork объект.

function out = resnet_predict(in)

persistent mynet;
if isempty(mynet)
    mynet = coder.loadDeepLearningNetwork('resnet50', 'myresnet');
end

out = predict(mynet,in);

Создайте coder.gpuConfig объект настройки для генерации кода MEX.

cfg = coder.gpuConfig('mex');

Установите выходной язык на C++. 

cfg.TargetLang = 'C++';

Создайте coder.CuDNNConfig объект настройки глубокого обучения и присвоение это к DeepLearningConfig свойство cfg объект настройки.

cfg.DeepLearningConfig = coder.DeepLearningConfig('cudnn');

Используйте -config опция codegen функционируйте, чтобы передать cfg объект настройки. codegen функция должна определить размер, класс и сложность входных параметров функции MATLAB®. Используйте -args опция, чтобы задать размер входа к функции точки входа.

codegen -args {ones(224,224,3,'single')} -config cfg resnet_predict;

codegen команда помещает все сгенерированные файлы в codegen папка. Папка содержит код CUDA для функции точки входа resnet_predict.cu, заголовок и исходные файлы, содержащие определения класса C++ для замысловатой нейронной сети (CNN), веса и файлов смещения.

Смотрите также

| | | | | | |

Темы

Введенный в R2018b

Документация GPU Coder
Поддержка