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

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

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

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

相关文章:

  • 2021-07-06
  • 2021-04-01
  • 2021-08-20
  • 2022-12-23
  • 2021-05-19
  • 2021-09-15
  • 2022-02-27
  • 2021-11-08
猜你喜欢
  • 2021-12-21
  • 2021-08-08
  • 2022-12-23
相关资源
相似解决方案