- 博客(3)
- 收藏
- 关注
RSS订阅原创 tensorflow 对GPU的配置
tf1配置import keras.backend.tensorflow_backend as KTFfrom tensorflow import set_random_seedconfig = tf.ConfigProto(allow_soft_placement=True)# 设置最大占有GPU不超过显存的80%config.gpu_options.per_process_gpu_memory_fraction = 0.8# GPU资源使用率,先少量分配，按需逐步增加...
2021-04-21 15:00:06
446
原创 机器学习之keras模型保存为pb文件
keras模型是依赖tensorflow框架的，在恢复模型之前还需要再定义一遍网络结构，这对于部署到生产环境来说非常不方便。而转换为pb文件，可以独立运行，任何语言都可以解析它，同时方便部署到tf serving上。本文提供以下两种转换方法。方法1（推荐）：适用于tf2.0之后的版本，但是1.0版本生成的hdf5文件也可以用此方法转换，前提必须在tf2.0环境下运行model_path = 'weights.hdf5' # 模型文件model =.
2021-04-16 10:08:39
3003
原创 python机器学习常用包
1、Numpy使用Python进行科学计算的基本软件包，是scikit-learn和SciPy等机器学习库的基础。链接：https://numpy.org/安装方式：pippip install numpycondaconda install numpy2、Pandas快速，强大，灵活且易于使用的开源数据分析和处理工具。链接：https://pandas.pydata.org/...
2021-04-14 16:13:02
4564
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人