首页 > 其他 > 详细

TensorRT优化过程中的dropout问题

时间:2019-03-25 13:10:41      阅读:208      评论:0      收藏:0      [点我收藏+]

使用tensorRT之前,你一定要注意你的网络结构是否能够得到trt的支持,无论是CNN还是RNN都会有trt的操作。

例如:tf.nn.dropout(features, keep_prob),trt就不支持。

这个也不奇怪,因为trt在要求输入中,只要你传入样本数据,那你就不能feed一个数值,所以以后直接想都不要想把keep_prob传进来。

为了克服过拟合,你可以使用tf.layers.dense()方法,在后期预测的时候,应该不用确定的这个值了,初步想法,等待验证。

TensorRT优化过程中的dropout问题

原文:https://www.cnblogs.com/demo-deng/p/10593009.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!