ProGamerGov

from collections import OrderedDict
from typing import Callable, Dict, Optional
from warnings import warn

import torch

def _remove_all_forward_hooks(
    module: torch.nn.Module, hook_fn_name: Optional[str] = None
) -> None:
    """