Korabelnikov Aleks korabelnikov

Machine Learning/Deep Learning/Computer Vision [email protected]

ProGamerGov / remove_hooks.py

Created May 1, 2022 18:23

Remove hooks in PyTorch without using the hook handle

	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:
	"""