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

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

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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐