Tensorflow
智障大人
你说你很累,但谁又过得顺风又顺水
展开
-
tensorflow GPU 报错
1.tensorflow GPU 报错2.报错语句Function call stack: distributed_function在开头添加如下代码physical_devices = tf.config.experimental.list_physical_devices('GPU')assert len(physical_devices) > 0, "Not enough GP...原创 2020-03-05 23:33:33 · 1195 阅读 · 0 评论 -
Tensorflow
Tensorflowtf.get_collection()主要作用从一个集合中取出变量tf.get_collection( key, scope=None)例子tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES, scope="hidden[123]")表示获取第1,2,3隐藏层的权重...原创 2020-02-07 17:27:08 · 184 阅读 · 0 评论