我有一个训练有素的模型,有10个课程,我需要添加新课程,而不会失去预训练的重量。我知道为了解决这个问题,我首先应该加载预训练,移除/替换FC,冻结预训练模型的下层并训练网络。所以理论是清楚但实际上我应该调用加载以前训练模型的方法,因为我只是在恢复最后一个检查点 saver.restore(sess,tf.train.latest_checkpoint)之间或者通过导入meta图表通过调用 tf.train.import_meta_graph 。
根据冻结所有重量,我不知道哪种方法对此负责?任何想法或是否有任何其他可能的方法来添加新类。