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})
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})