深度学习
chenyu19880302
这个作者很懒,什么都没留下…
展开
-
windows10-cuda10.1-cudnn7-tensorflow1.4.0配置流程
1.下载cuda版本:cuda_10.1.105_win10_network.exe,并按照流程安装。安装完后,打开cmd ,输入:nvcc -V,回车查看是否有版本信息,若出现版本信息,则证明nvcc安装成功。2.下载cudnn版本:cudnn-10.0-windows10-x64-v7.4.2.24,并把bin,include,lib三个文件拷贝到CUDA/10.1文件夹中。3.下载A...原创 2019-03-23 19:10:22 · 2568 阅读 · 0 评论 -
tensorflow加载预训练模型的部分参数方法
在训练时,修改了部分网络结构,但其他的网络结果仍一样,又不想重新开始训练,这就需要加载已训练的好的模型的参数。下面是在训练中需要加载部分参数模型,以此记录:if args.checkpoint and os.path.isdir(args.checkpoint): logger.info('Restore from checkpoint...') ...原创 2019-05-24 15:46:04 · 8538 阅读 · 0 评论