Source code for draugr.torch_utilities.writers.torch_module_writer.module_parameter_writer_mixin
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from abc import ABC, abstractmethod
__author__ = "Christian Heider Nielsen"
__doc__ = r"""
Created on 09/10/2019
"""
__all__ = ["ModuleParameterWriterMixin"]
import torch
[docs]class ModuleParameterWriterMixin(ABC):
"""
Writer mixin that provides an interface for 'writing' torch Module parameters"""
[docs] @abstractmethod
def parameters(self, tag: str, model: torch.nn.Module, step: int, **kwargs) -> None:
"""
:param tag:
:type tag:
:param model:
:type model:
:param step:
:type step:
:param kwargs:
:type kwargs:"""
raise NotImplementedError