深度学习
绿竹巷人
不想做算法工程师的算法工程师
展开
-
onnx模型转为trt模型
想要加速推理速度,一定要用半精度/混合精度,即(-d 16)原创 2022-10-28 14:44:24 · 1968 阅读 · 0 评论 -
Ubuntu20.04安装torch
根据python版本安装对应的pytorch。原创 2022-10-10 14:42:19 · 4028 阅读 · 0 评论 -
tensorflow实战(二)——自编码器
自编码器自行百度,待补充学习1. one_hot 独热编码mnist = input_data.read_data_sets('MNIST_data/', one_hot = True)可以把数据(比如5以内的数)编码成5维,数据在哪,哪个位置就是1的 [0, 0, 0, 0, 1]这种数据格式2. 数据标准化处理preprocessor = prep.StandardScaler...原创 2020-04-16 18:57:27 · 264 阅读 · 0 评论 -
tensorflow实战(一)——MNIST数据集的hello world
描述这一系列文章全部参考黄文坚 的《tensorflow实战》第三章学习交叉熵cross_entropy = tf.reduce_mean(-tf.reduce_sum(y_ * tf.log(y), reduction_indices=[1]))reduction_indices=[0]的意思是,按行求和(列数不变,行全部加在一起)reduction_indices=[1]的意...原创 2020-04-16 11:13:44 · 121 阅读 · 0 评论 -
tensorflow运行时的一些警告提示解决方法
描述运行tensorflow程序时,跳出的一些警告提示的解决办法持续更新问题1FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)ty...原创 2020-04-10 16:35:13 · 2227 阅读 · 0 评论 -
tensorflow1和2的mnist代码
描述tensorflow1和2的具体代码是有显著差异的,这篇文章会分别写出两个版本的mnist手写体识别代码这里先记录一句命令当你安装了tensorflow2却仍想按照tensorflow1的API来编写的话,最开始加上这样一句import tensorflow.compat.v1 as tftensorflow 1代码#encoding: utf-8import tenso...原创 2020-03-30 21:20:07 · 590 阅读 · 0 评论 -
Mac安装不同版本的tensorflow
问题描述Mac电脑安装tensorflow的步骤及安装中出现的问题首先来说tensorflow版本tensorflow的版本现在有两种,tensorflow1和tensorflow2。应该说两者在代码上的差别还挺大的,具体这里不做列举(我其实也没去看有多少不同,有兴趣的前往tensorflow的github官网学习)自然,我们就有了两个选择安装最新的tensorflow,也就是tens...原创 2020-03-30 10:12:06 · 2622 阅读 · 2 评论