-1
我想「手動」操作TF slim保存的模型檢查點的權重。我需要訓練一個新版本的纖細模型,以進行可視化和分佈式訓練。有沒有辦法修改檢查點中保存的隨機初始權重並使用該檢查點進行訓練?更改TF檢查點的權重以用Slim訓練
我想「手動」操作TF slim保存的模型檢查點的權重。我需要訓練一個新版本的纖細模型,以進行可視化和分佈式訓練。有沒有辦法修改檢查點中保存的隨機初始權重並使用該檢查點進行訓練?更改TF檢查點的權重以用Slim訓練
definitly你可以看到這個answer參考
摘要:
你首先需要它的字符串映射到numpy.array字典,字符串是將分配張名到一個變量,並運行所有variable.assign
ops與此字典作爲feed_dict sess.run