Перейти к содержимому
Compvision.ru
BeS

Переиспользование частей нейронных сетей в caffe

Recommended Posts

Всем привет,

А кто-то сталкивался с задачей переиспользования кусков сетей в caffe? Например, я хочу использовать часть сверточных слоев от VGG16 при тренировке своей кастомной сети, как именно это технически реализуется? Тупо копипаста из одного caffemodel в другой весов не работает...caffe такие файлы просто игнорит.

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Я не использовал, но в версии 3.1 упоминается:

DIGITS 3.1

@lukeyeager lukeyeager released this  · 22 commits to master since this release

New Features

  • Enable multi-GPU for Torch (#480)
  • Add basic login functionality (#463)
  • Allow Torch to fine-tune pretrained models (#499)
  • Allow Caffe to fine-tune from multiple pretrained models (#498)
  • New tutorials
    • Fine-tuning (#500)
    • Siamese networks (#453)
    • Weight initialization (#522)
  • Allow optional specification of image folder during multiple inference (#526)

Есть подозрение что это близко к тому что Вы ищите.

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Очень даже любопытно, неужто они втащили какие-то методы по компрессии сеток за счет предтренировки внутрь своего DIGITS'а

Поделиться сообщением


Ссылка на сообщение
Поделиться на других сайтах

Создайте учётную запись или войдите для комментирования

Вы должны быть пользователем, чтобы оставить комментарий

Создать учётную запись

Зарегистрируйтесь для создания учётной записи. Это просто!

Зарегистрировать учётную запись

Войти

Уже зарегистрированы? Войдите здесь.

Войти сейчас


  • Сейчас на странице   0 пользователей

    Нет пользователей, просматривающих эту страницу

×