TensorRT优化过程中的dropout问题

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

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

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

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

猜你喜欢

转载自www.cnblogs.com/demo-deng/p/10593009.html
今日推荐