Function trainable_params

tools/train_utils/optimization/fastai_optim.py:94–97

Return list of trainable params in `m`.

(m: nn.Module)

def trainable_params(m: nn.Module):
    "Return list of trainable params in `m`."
    res = filter(lambda p: p.requires_grad, m.parameters())
    return res


def is_tuple(x) -> bool: return isinstance(x, tuple)