在ScipyOptimizeDriver 中,并非所有优化器都支持梯度优化,因此在设置驱动程序之前您无法确定正确的值。这是在您的问题的final_setup() 中完成的(在您的驱动程序中调用_setup_driver())。此方法在run_model() 和run_driver() 中调用,但您也可以单独调用它来获取优化器的正确属性。
在下面的示例中,我询问了驱动程序 3 次是否支持渐变。第一次,在 problem 设置之后它给出了一个错误的答案(默认),因为驱动程序还没有被触及。如果我调用final_setup(),这将设置驱动程序,并且驱动程序的所有属性都是正确的。如果调用run_model() 或run_driver(),当然这也会设置驱动程序。
所以我的建议是在从驱动程序中查询任何内容之前使用final_setup(),这可能会在设置过程中发生变化(主要是优化器特定的属性)。
import openmdao.api as om
# build the model
prob = om.Problem()
indeps = prob.model.add_subsystem('indeps', om.IndepVarComp())
indeps.add_output('x', 3.0)
indeps.add_output('y', -4.0)
prob.model.add_subsystem('paraboloid', om.ExecComp('f = (x-3)**2 + x*y + (y+4)**2 - 3'))
prob.model.connect('indeps.x', 'paraboloid.x')
prob.model.connect('indeps.y', 'paraboloid.y')
# setup the optimization
driver = prob.driver = om.ScipyOptimizeDriver()
prob.driver.options['optimizer'] = 'SLSQP'
prob.model.add_design_var('indeps.x', lower=-50, upper=50)
prob.model.add_design_var('indeps.y', lower=-50, upper=50)
prob.model.add_objective('paraboloid.f')
prob.setup()
print("\nSupports gradients (after setup)?")
print(prob.driver.supports['gradients'])
prob.final_setup()
print("\nSupports gradients (after final setup)?")
print(prob.driver.supports['gradients'])
prob.run_driver()
print("\nSupports gradients (after run)?")
print(prob.driver.supports['gradients'])
这会产生以下输出:
Supports gradients (after setup)?
False
Supports gradients (after final setup)?
True
Optimization terminated successfully. (Exit mode 0)
Current function value: -27.33333333333333
Iterations: 5
Function evaluations: 6
Gradient evaluations: 5
Optimization Complete
-----------------------------------
Supports gradients (after run)?
True