TensorRT:BatchNorm

首先學習一下pytorch中的BN層: BN層的參數有weight, bias, running_mean, running_var, num_batches_tracked  BN的公式如下圖: 很顯然, running_mean 是均值, running_var 是方差, 那gamma、beta去哪裏了? gamma、beta是可學習的參數,weight 和 bias也是可學習參數,實際上:w
本站公眾號
   歡迎關注本站公眾號,獲取更多信息