回覆列表
  • 1 # 使用者5231388881052

    可以load model之後將不需要的layer置為nil,例如將最後一層卷積之後的層去掉:require "loadcaffe"local net = loadcaffe.load("×/VGG_ILSVRC_16_layers_deploy.prototxt",×/VGG_ILSVRC_16_layers.caffemodel","nn")--print(net) --此處看到的網路有40層for i = 30,40 do net.modules[i] =nilendprint(net) -- 此處的網路最後一層卷積層後面的沒有了---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------更官方的方式為:for i = 40,30,-1 do net:remove(i)end

  • 中秋節和大豐收的關聯?
  • 天地蒼蒼,乾坤茫茫,中華少年,頂天立地當自強,出自哪裡?