Transfer learning with tf.estimator.Estimator framework

前端 未结 2 507
独厮守ぢ
独厮守ぢ 2021-02-04 05:04

I\'m trying to do transfer learning of an Inception-resnet v2 model pretrained on imagenet, using my own dataset and classes. My original codebase was a modification of a

2条回答
  •  囚心锁ツ
    2021-02-04 05:54

    I find out {s+'/':s+'/' for s in scopes} didn't work, just because the variables_to_restore include something like "global_step", so scopes include the global scopes which could include everything. You need to print variables_to_restore, find "global_step" thing, and put it in "exclude".

提交回复
热议问题