- 博客(7)
- 收藏
- 关注
RSS订阅原创 Torch使两个Dataloader保持一致的打乱顺序
PYG中，数据类无法自定义，有时候要汇聚多组图数据，此时要求Dateloader的打乱顺序一致使得他们输出相同。
2024-09-03 19:44:06
193
原创 Numpy中如何输出Array前N最大值的索引？
我们使用arr.argsort() 函数实现这个功能top_k=3arr = np.array([2,5,12,6,1])top_k_idx=arr.argsort()[::-1][0:top_k]print(top_k_idx)接下来我们将之封装为函数:def ArrayTopK(top_k,arr) top_k_idx=arr.argsort()[::-1][0:top_k] return top_k_idx...
2022-04-02 19:47:08
840
原创 Tensorflow 中如何修改模型的权重？
首先可以使用get_weights() API获取权值。t=watermark_model.layers[1].get_weights()修改t（这个变量的类型大家就自己看啦)t[0][0,0,0,0]=0使用 set_weights(weight) API 设置权重。watermark_model.layers[1].set_weights(t)可以看见已经发生改变。...
2022-04-02 19:40:26
3496
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人