作者:会员WWW | 来源:互联网 | 2023-08-19 10:28
#Initializetheweightsbiaswithidentitytransformationself.fc_loc[2].weight.data.zero_()sel
# Initialize the weights/bias with identity transformation
self.fc_loc[2].weight.data.zero_()
self.fc_loc[2].bias.data.copy_(torch.tensor([1, 0, 0, 0, 1, 0], dtype=torch.float))
或者用apply函数执行批量初始化