def load(self, numpy_dict):
        """
            Load data from a numpy_dictionary

        Args:
            numpy_dict: {
                <module_dict_key>: {
                    <paramter_key_module_dict_entries>: np.array
                }
            }
        """
        for key in numpy_dict:
            for k in numpy_dict[key]:
                numpy_dict[key][k] = torch.from_numpy(
                    np_cast(numpy_dict[key][k], np.float32))
        for k, m in self._module_dict.items():
            m.load_state_dict(numpy_dict[k])

            Load data from a numpy_dictionary

        Args:
            numpy_dict: {
                <module_dict_key>: {
                    <paramter_key_module_dict_entries>: np.array
                }
            }
        
surreal/distributed/module_dict.py