TensorFlow
正门大石狮
写可以复现的实验,写可以看懂的理论。
展开
-
【BUG】Keras用了multi_gpu_model,但用Nvidia-smi看只用了一块显卡GPU
解决方法很简单,在你原来使用multi_gpu_model外层加上两句代码:strategy = tf.distribute.MirroredStrategy(devices=["/gpu:0","/gpu:1", "/gpu:2"]) with strategy.scope():原创 2020-11-25 12:36:17 · 1629 阅读 · 8 评论 -
TensorFlow中的tf.slice()是个什么操作
tf.slice() 定义如下 tf.slice( input_, begin, size, name=None ) 参数 意义 input_ 一个Tensor begin 各个维度的起始坐标组成的向量,从0开始 size 各个维度取出来的值的个数所组成的向量,从1开始 示例 一个Tensor如下所示:其每个维度上值的个数分别是3、2、3个(由外到内)。 t ...原创 2020-04-28 15:53:40 · 482 阅读 · 0 评论