91超碰碰碰碰久久久久久综合_超碰av人澡人澡人澡人澡人掠_国产黄大片在线观看画质优化_txt小说免费全本

溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

tensorflow 恢復指定層與不同層指定不同學習率的方法

發布時間:2020-08-20 16:03:19 來源:腳本之家 閱讀:188 作者:跬步達千里 欄目:開發技術

如下所示:

#tensorflow 中從ckpt文件中恢復指定的層或將指定的層不進行恢復:
#tensorflow 中不同的layer指定不同的學習率
 
with tf.Graph().as_default():
		#存放的是需要恢復的層參數
	 variables_to_restore = []
	 #存放的是需要訓練的層參數名,這里是沒恢復的需要進行重新訓練,實際上恢復了的參數也可以訓練
  variables_to_train = []
  for var in slim.get_model_variables():
   excluded = False
   for exclusion in fine_tune_layers:
   #比如fine tune layer中包含logits,bottleneck
    if var.op.name.startswith(exclusion):
     excluded = True
     break
   if not excluded:
    variables_to_restore.append(var)
    #print('var to restore :',var)
   else:
    variables_to_train.append(var)
    #print('var to train: ',var)
 
 
  #這里省略掉一些步驟,進入訓練步驟:
  #將variables_to_train,需要訓練的參數給optimizer 的compute_gradients函數
  grads = opt.compute_gradients(total_loss, variables_to_train)
  #這個函數將只計算variables_to_train中的梯度
  #然后將梯度進行應用:
  apply_gradient_op = opt.apply_gradients(grads, global_step=global_step)
  #也可以直接調用opt.minimize(total_loss,variables_to_train)
  #minimize只是將compute_gradients與apply_gradients封裝成了一個函數,實際上還是調用的這兩個函數
  #如果在梯度里面不同的參數需要不同的學習率,那么可以:
 
  capped_grads_and_vars = []#[(MyCapper(gv[0]), gv[1]) for gv in grads_and_vars]
  #update_gradient_vars是需要更新的參數,使用的是全局學習率
  #對于不是update_gradient_vars的參數,將其梯度更新乘以0.0001,使用基本上不動
 	for grad in grads:
 		for update_vars in update_gradient_vars:
 			if grad[1]==update_vars:
 				capped_grads_and_vars.append((grad[0],grad[1]))
 			else:
 				capped_grads_and_vars.append((0.0001*grad[0],grad[1]))
 
 	apply_gradient_op = opt.apply_gradients(capped_grads_and_vars, global_step=global_step)
 
 	#在恢復模型時:
 
  with sess.as_default():
 
   if pretrained_model:
    print('Restoring pretrained model: %s' % pretrained_model)
    init_fn = slim.assign_from_checkpoint_fn(
    pretrained_model,
    variables_to_restore)
    init_fn(sess)
   #這樣就將指定的層參數沒有恢復

以上這篇tensorflow 恢復指定層與不同層指定不同學習率的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。

向AI問一下細節

免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI

井陉县| 广饶县| 千阳县| 广灵县| 马尔康县| 平武县| 常山县| 永福县| 鹰潭市| 鄂托克前旗| 鄱阳县| 阜新| 长兴县| 建水县| 左权县| 滕州市| 庆云县| 十堰市| 卫辉市| 杨浦区| 鹤岗市| 育儿| 同仁县| 罗甸县| 丹江口市| 鹤峰县| 藁城市| 天门市| 泽普县| 合江县| 修水县| 左云县| 尉氏县| 夏河县| 张家界市| 秦皇岛市| 洪湖市| 陆良县| 两当县| 红安县| 新龙县|