Source code for inFairness.utils.params

import torch.nn



[docs]
def freeze_network(network: torch.nn.Module):
    """Freeze network parameters.
    :param network: torch.nn.Module
    :type network: torch.nn.Module
    """
    for p in network.parameters():
        p.requires_grad = False




[docs]
def unfreeze_network(network: torch.nn.Module):
    """Unfreeze network parameters.
    :param network: torch.nn.Module
    :type network: torch.nn.Module
    """
    for p in network.parameters():
        p.requires_grad = True