|
|
@ -316,6 +316,7 @@ class TestQuantizationFreezePass(unittest.TestCase):
|
|
|
|
build_strategy = fluid.BuildStrategy()
|
|
|
|
build_strategy = fluid.BuildStrategy()
|
|
|
|
build_strategy.memory_optimize = False
|
|
|
|
build_strategy.memory_optimize = False
|
|
|
|
build_strategy.enable_inplace = False
|
|
|
|
build_strategy.enable_inplace = False
|
|
|
|
|
|
|
|
build_strategy.fuse_all_reduce_ops = False
|
|
|
|
binary = fluid.CompiledProgram(main_graph.graph).with_data_parallel(
|
|
|
|
binary = fluid.CompiledProgram(main_graph.graph).with_data_parallel(
|
|
|
|
loss_name=loss.name, build_strategy=build_strategy)
|
|
|
|
loss_name=loss.name, build_strategy=build_strategy)
|
|
|
|
quantized_test_program = test_graph.to_program()
|
|
|
|
quantized_test_program = test_graph.to_program()
|
|
|
|