Method load_state_dict

code/deep/CSG/methods/supvae.py:84–86

(self, state_dict: dict)

        return self._parameter_dict

    def load_state_dict(self, state_dict: dict):
        for name in list(self._parameter_dict.keys()):
            with tc.no_grad(): self._parameter_dict[name].copy_(state_dict[name])

    def get_lossfn(self, n_mc_q: int=0, reduction: str="mean", mode: str="defl", weight_da: float=None, wlogpi: float=None):
        if reduction == "mean": reducefn = tc.mean