BOAT-MS Structure

Core Problem Class

class boat_ms.boat_opt.Problem(config, loss_config)

Bases: object

Enhanced bi-level optimization problem class supporting flexible loss functions and operation configurations.

build_ll_solver()

Configure the lower-level solver.

build_ul_solver()

Placeholder for UL solver configuration (FOGO manages updates internally in MindSpore setting).

plot_losses()

run_iter(ll_feed_dict, ul_feed_dict, current_iter)

Run a single iteration of the bi-level optimization process.

Returns:

(results, run_time)

save_losses(current_iter, ll_loss, ul_loss)

Save the losses to a JSON file and update the loss history.

Main Subpackages

• boat_ms.gm_ol
  • Submodules
  • boat_ms.gm_ol.dynamical_system
    • DynamicalSystem
      • DynamicalSystem.optimize()
• boat_ms.fo_ol
  • Submodules
  • boat_ms.fo_ol.meso
    • MESO
      • MESO.optimize()
  • boat_ms.fo_ol.pgdo
    • PGDO
      • PGDO.optimize()
  • boat_ms.fo_ol.vfo
    • VFO
      • VFO.optimize()
  • boat_ms.fo_ol.vso
    • VSO
      • VSO.optimize()
• boat_ms.utils
  • Submodules
  • boat_ms.utils.op_utils
    • average_grad()
    • copy_parameter_from_list()
    • grad_unused_zero()
    • l2_reg()
    • require_model_grad()
    • stop_grads()
    • stop_model_grad()
    • update_grads()
    • update_tensor_grads()

Extension with Operation_Registry

boat_ms.operation_registry.get_registered_operation(name)

Retrieve a registered operation class by name.

boat_ms.operation_registry.register_class(cls)

Register a new operation class to the global registry.