Сгенерированная иерархия классов CNN
Сгенерированный код CNN имеет следующую иерархию классов. Класс Слоя и сгенерированный Сетевой класс имеют три важных метода:
setup(), который выделяет ресурсы памяти и системные ресурсы для каждого слоя.predict(), который выполняет прямой вывод в цикле выполнения.cleanup(), который выпускает все ресурсы памяти и системные ресурсы.
Похожие темы
- Поддерживаемые сети и слои
- Загрузите предварительно обученные сети для генерации кода
- Генерация кода для Нейронных сетей для глубокого обучения при помощи cuDNN
- Генерация кода для нейронных сетей для глубокого обучения при помощи TensorRT
- Генерация кода для нейронных сетей для глубокого обучения, предназначающихся для ARM Мали графические процессоры