Source code for draugr.torch_utilities.optimisation.scheduling.learning_rate_utilities
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = "Christian Heider Nielsen"
__doc__ = r"""
Created on 20/10/2019
"""
__all__ = ["set_lr"]
[docs]def set_lr(optimizer, lr):
"""
:param optimizer:
:type optimizer:
:param lr:
:type lr:"""
for param_group in optimizer.param_groups:
param_group["lr"] = lr