Search code examples
deep-learningcaffeconv-neural-networkpycaffenvidia-digits

Multiple pretrained networks in Caffe


Is there a simple way (e.g. without modifying code) to load wights from multiple pretrained networks into one network? The network contains some layers with same dimensions and names as both pretrained networks.

I am trying to achieve this using NVidia DIGITS and Caffe.

EDIT: I thought it wouldn't be possible to do it directly from DIGITS, as confirmed by answers. Can anyone suggest a simple way to modify the DIGITS code to be able to select multiple pretrained networks? I checked the code a bit, and thought the training script would be a good place to start, but I don't have in-depth knowledge of Caffe, so I'm not sure what the best/quickest way to achieve this would be.


Solution

  • As Shai suggested, there was no way of doing this, so I decided to clone the official repository and make the appropriate changes. I changed the code so that multiple pretrained networks can be loaded by using a colon as separator.

    I created a pull request on the official repository and my changes were then merged with the main branch of DIGITS, meaning it is now possible to use this functionality in DIGITS.