optimization_pass.cc:185] None of the MLIR Optimization Passes are enabled (registered 2)解决办法

I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:185] None of the MLIR Optimization Passes are enabled (registered 2)

主要原因,训练使用内存或显存不足:

1、修改:Batch大小。
2、修改每个加载数据大小
3、减少训练网络复杂程度,减少训练网络涉及到得参数数量,减少训练网络的步骤等。

你可能感兴趣的:(tensorflow,tensorflow,深度学习,人工智能)