Search in sources :

Example 1 with MatrixSaveContextProto

use of com.tencent.angel.protobuf.generated.ClientMasterServiceProtos.MatrixSaveContextProto in project angel by Tencent.

the class ProtobufUtil method convert.

public static ModelSaveContext convert(ModelSaveContextProto saveContext) {
    ModelSaveContext context = new ModelSaveContext(saveContext.getSavePath());
    context.setCheckpoint(saveContext.getCheckpint());
    List<MatrixSaveContextProto> matricesContext = saveContext.getMatrixContextesList();
    int size = matricesContext.size();
    for (int i = 0; i < size; i++) {
        context.addMatrix(convert(matricesContext.get(i)));
    }
    return context;
}
Also used : MatrixSaveContextProto(com.tencent.angel.protobuf.generated.ClientMasterServiceProtos.MatrixSaveContextProto) PSMatrixSaveContextProto(com.tencent.angel.protobuf.generated.PSMasterServiceProtos.PSMatrixSaveContextProto) ModelSaveContext(com.tencent.angel.model.ModelSaveContext)

Aggregations

ModelSaveContext (com.tencent.angel.model.ModelSaveContext)1 MatrixSaveContextProto (com.tencent.angel.protobuf.generated.ClientMasterServiceProtos.MatrixSaveContextProto)1 PSMatrixSaveContextProto (com.tencent.angel.protobuf.generated.PSMasterServiceProtos.PSMatrixSaveContextProto)1