Syntax with clip_args for pyro.optim.LambdaLR

Hi, looks like you need to pass clip_args as a separate keyword argument to pyro.optim.LambdaLR:

pyro.optim.LambdaLR({'optimizer': optimizer, 'optim_args': {...}, 'lr_lambda': lambda1},
                    clip_args={"clip_norm":10.0})
1 Like