Jump to content
Compvision.ru
Sign in to follow this  
BeS

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

Recommended Posts

Всем привет,

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

Share this post


Link to post
Share on other sites

Я не использовал, но в версии 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)

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

Share this post


Link to post
Share on other sites

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

Share this post


Link to post
Share on other sites

Create an account or sign in to comment

You need to be a member in order to leave a comment

Create an account

Sign up for a new account in our community. It's easy!

Register a new account

Sign in

Already have an account? Sign in here.

Sign In Now

Sign in to follow this  

  • Recently Browsing   0 members

    No registered users viewing this page.

×