LCCFlccf的博客

浅谈深度学习:如何计算模型以及中间变量的显存占用大小：https://oldpan.me/archives/how-to-calculate-gpu-memory如何在Pytorch中精细化利用显存https://oldpan.me/archives/how-to-use-memory-pytorchpytorch 减小显存消耗，优化显存使用，避免out of memory...

在用tensorflow搭建好网络之后，如果可视化一下网络的结构与变量，会对网络结构有一个更直观的了解。网络结构与变量的可视化方法如下：在搭建好网络之后，如下代码可以打印出网络的变量variables = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)for v in variables: print(v)活动的...