深度学习
Longriver111
新晋AI攻城狮
展开
-
tensorflow 常用函数整理
tensorflow常用函数整理1 tf.reshape2 tf.concat()3.tf.expand_dims 记录自己使用过的tensorflow常用函数,以便以后查询 1 tf.reshape tf.reshape( tensor, shape, name=None ) 将给定tensor的维度转换为shape。 arr = tf.Variable([[1,2],...原创 2019-05-15 21:38:45 · 142 阅读 · 0 评论 -
解决tensorflow在训练时默认占用所有GPU的显存
tensorflow在训练时默认占用所有GPU的显存。 可以通过以下方式解决该问题: 1.在构造tf.Session()时候通过传递tf.GPUOptions作为可选配置参数的一部分来显式地指定需要分配的显存比例,如下所示: # 假如有12GB的显存并使用其中的4GB: gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.33...转载 2019-06-26 21:53:31 · 3791 阅读 · 0 评论 -
xlnet pytorch简易版代码解读
xlnet-pytorch版本代码解读安装参数分析data_utils.py代码分析 这篇不详细介绍xlnet的原理,有需要了解的请自行前往xlnet原理分析 安装 首先clone XLNet-pytorch的源码 git clone https://github.com/graykode/xlnet-Pytorch && cd xlnet-Pytorch # To use Se...翻译 2019-07-19 14:49:06 · 2325 阅读 · 1 评论