Source code for draugr.torch_utilities.persistence.naming

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 02-12-2020
           """

__all__ = ["get_model_hash"]

import hashlib

import torch


[docs]def get_model_hash(model: torch.nn.Module) -> str: """description""" model_repr = "".join([str(a) for a in model.named_children()]) # print(model_repr) model_hash = hashlib.md5(model_repr.encode("utf-8")).hexdigest() return model_hash