2016-05-10 3 views
1

あらかじめ訓練されたニューラルネットを微調整したい。しかし、このモデルはCaffe製で、Torchで働きたいです。
私はloadcaffeを試しましたが、これはfinetuningに焦点を当てていないようです。
これを可能にする別のツールがありますか?あるいは、CaffeモデルをTorchネットに変換できますか?トーチのカフェモデルファインチューン

答えて

1

あなたがやらなければならないことは、トーチのバージョンにあなたのカフェpretrainedネットワークを変換するため

  1. 使用loadcaffeです。
    (直後にディスクに保存することができますtorch.save("net.t7", model)
  2. トーチコードを微調整してください。
+0

ちょうど私が必要としたもの。ありがとう! – Cassie

+0

@Cassieはこの仕事をしましたか? –

関連する問題