examples/research_projects/seq2seq-distillation/utils.py/assert_not_all_frozen
def assert_not_all_frozen(model):
    model_grads: List[bool] = list(grad_status(model))
    npars = len(model_grads)
    assert any(model_grads), f"none of {npars} weights require grad"