iteye_7016的博客

pytorch保存模型、参数的方法方法一：只保存模型的参数 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu' ) filepath = 'model.dat' # 保存参数 torch.save(model.state_dict(), filepath) # 加载模型参数 , map_location: 把数据加载到哪个device（GPU或CPU） model.load_stat

@tensorflow.reduce_函数tensorflow.reduce_sum()reduce_sum函数的计算和指定维度的顺序无关a = np.random.randint(1, 9, size=(2,3,4))print(a)b1 = tf.reduce_sum(tf.convert_to_tensor(value=a),axis=[2,1],keepdims=True)print(b1) b2 = tf.reduce_sum(tf.convert_to_tensor(value