If one sets max_guidance_updates: 0 in the configuration of the lambo optimizer, a silly error pops up:
Exception has occurred: UnboundLocalError
local variable 'design_loss' referenced before assignment
Which happens in line 215 of _lambo.py, building a metrics dict for wandb logging.
This problem can be solved by slightly modifying the for lang_step in range(self.max_guidance_updates) for-loop, adding some default Nones for the losses.
If one sets
max_guidance_updates: 0
in the configuration of thelambo
optimizer, a silly error pops up:Which happens in line 215 of
_lambo.py
, building a metrics dict forwandb
logging.This problem can be solved by slightly modifying the
for lang_step in range(self.max_guidance_updates)
for-loop, adding some defaultNone
s for the losses.