Python ScriptModule 예제들

예제 #1

    def bufferize(worker: AbstractWorker, script_module: torch.jit.ScriptModule) -> ScriptModulePB:
        """
            This method serializes a torch.jit.ScriptModule using ScriptModulePB.

            Args:
                script_module (torch.jit.ScriptModule): input jit.ScriptModule to be serialized.

            Returns:
                protobuf_script (ScriptModulePB): serialized jit.ScriptModule.
        """
        protobuf_script = ScriptModulePB()
        protobuf_script.obj = script_module.save_to_buffer()
        return protobuf_script

예제 #2

파일: torch_serde.py 프로젝트: zeta1999/PySyft

def _bufferize_traced_module(
        worker: AbstractWorker,
        script_module: torch.jit.TopLevelTracedModule) -> TracedModulePB:
    protobuf_script = ScriptModulePB()
    protobuf_script.obj = script_module.save_to_buffer()
    return protobuf_script