TensorFlow
AlanBupt
种一颗树最好的时间是十年前,其次是现在
展开
-
TensorFlow中padding的 'SAME' 和 'VALID' 参数
VALIDconv2d中VALID参数不会在输入上添加新的像素,输出图片的宽度的计算公式为output_width = [(input_width - kernel_size + 1) / stride]其中[ ]表示结果向上取整,输出图像的宽度计算类似SAMEconv2d中SAME参数会填充0,确保卷积结果覆盖输入图像的所有像素点,计算宽度的公式为outp...原创 2018-03-27 22:47:26 · 393 阅读 · 0 评论 -
关于TensorFlow的GPU设置
在使用GPU版的TensorFlow跑程序的时候,如果不特殊写代码注明,程序默认是占用所有主机上的GPU,但计算过程中只会用其中一块。也就是你看着所有GPU都被占用了,以为是在GPU并行计算,但实际上只有其中一块在运行;另外的所有显卡都闲着,但其显存都被占用了,所以别人也用不了。不过这种情况通过在程序之前加三行代码就可以解决:import osos.environ["CUDA_DEVIC...原创 2018-04-04 00:35:11 · 15487 阅读 · 1 评论