!14563 delete trainModel class
From: @HilbertDavid Reviewed-by: @zhanghaibo5,@ddwsky Signed-off-by: @zhanghaibo5pull/14563/MERGE
mindspore-ci-bot
4 years ago
committed by
Gitee
commit
73dfefea1c
@ -1,91 +0,0 @@
|
|||||||
/**
|
|
||||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
#include "src/train/train_model.h"
|
|
||||||
#include "src/common/log_adapter.h"
|
|
||||||
#include "include/errorcode.h"
|
|
||||||
#include "src/common/graph_util.h"
|
|
||||||
|
|
||||||
namespace mindspore::lite {
|
|
||||||
|
|
||||||
TrainModel *TrainModel::Import(const char *model_buf, size_t size) {
|
|
||||||
if (model_buf == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "The model buf is nullptr";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
TrainModel *model = new (std::nothrow) TrainModel();
|
|
||||||
if (model == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "new model fail!";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
model->buf = reinterpret_cast<char *>(malloc(size));
|
|
||||||
if (model->buf == nullptr) {
|
|
||||||
delete model;
|
|
||||||
MS_LOG(ERROR) << "malloc inner model buf fail!";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
memcpy(model->buf, model_buf, size);
|
|
||||||
model->buf_size_ = size;
|
|
||||||
auto status = model->ConstructModel();
|
|
||||||
if (status != RET_OK) {
|
|
||||||
MS_LOG(ERROR) << "construct model failed.";
|
|
||||||
delete model;
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
return model;
|
|
||||||
}
|
|
||||||
|
|
||||||
void TrainModel::Free() {}
|
|
||||||
|
|
||||||
char *TrainModel::ExportBuf(char *buffer, size_t *len) const {
|
|
||||||
if (len == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "len is nullptr";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
if (buf_size_ == 0 || buf == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "Model::Export is only available for Train Session";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
if (*len < buf_size_ && buffer != nullptr) {
|
|
||||||
MS_LOG(ERROR) << "Buffer is too small, Export Failed";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
if (buffer == nullptr) {
|
|
||||||
buffer = reinterpret_cast<char *>(malloc(buf_size_));
|
|
||||||
if (buffer == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "allocated model buf fail!";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
memcpy(buffer, buf, buf_size_);
|
|
||||||
*len = buf_size_;
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
char *TrainModel::GetBuffer(size_t *len) const {
|
|
||||||
if (len == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "len is nullptr";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
if (buf_size_ == 0 || buf == nullptr) {
|
|
||||||
MS_LOG(ERROR) << "Model::Export is only available for Train Session";
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
*len = buf_size_;
|
|
||||||
return buf;
|
|
||||||
}
|
|
||||||
} // namespace mindspore::lite
|
|
||||||
@ -1,57 +0,0 @@
|
|||||||
/**
|
|
||||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
#ifndef MINDSPORE_LITE_SRC_TRAIN_TRAIN_MODEL_H_
|
|
||||||
#define MINDSPORE_LITE_SRC_TRAIN_TRAIN_MODEL_H_
|
|
||||||
#include <vector>
|
|
||||||
#include "src/lite_model.h"
|
|
||||||
|
|
||||||
namespace mindspore {
|
|
||||||
namespace lite {
|
|
||||||
/// \brief TrainModel Defines a class that allows to import and export a mindsport trainable model
|
|
||||||
struct TrainModel : public lite::LiteModel {
|
|
||||||
/// \brief Static method to create a TrainModel object
|
|
||||||
///
|
|
||||||
/// \param[in] model_buf A buffer that was read from a MS model file
|
|
||||||
/// \param[in] size Length of the buffer
|
|
||||||
//
|
|
||||||
/// \return Pointer to MindSpore Lite TrainModel
|
|
||||||
static TrainModel *Import(const char *model_buf, size_t size);
|
|
||||||
|
|
||||||
/// \brief Free meta graph related data
|
|
||||||
void Free() override;
|
|
||||||
|
|
||||||
/// \brief Class destructor, free all memory
|
|
||||||
virtual ~TrainModel() = default;
|
|
||||||
|
|
||||||
/// \brief Export Model into a buffer
|
|
||||||
///
|
|
||||||
/// \param[in] buf The buffer to Export into. If equal to nullptr, buf will be allocated
|
|
||||||
/// \param[in,out] len Size of the pre-allocated buffer, and returned size of the exported buffer
|
|
||||||
///
|
|
||||||
/// \return Pointer to buffer with exported model
|
|
||||||
char *ExportBuf(char *buf, size_t *len) const;
|
|
||||||
|
|
||||||
/// \brief Get Model buffer
|
|
||||||
///
|
|
||||||
/// \param[in,out] len Return size of the buffer
|
|
||||||
///
|
|
||||||
/// \return Pointer to model buffer
|
|
||||||
char *GetBuffer(size_t *len) const;
|
|
||||||
};
|
|
||||||
} // namespace lite
|
|
||||||
} // namespace mindspore
|
|
||||||
|
|
||||||
#endif // MINDSPORE_LITE_SRC_TRAIN_TRAIN_MODEL_H_
|
|
||||||
Loading…
Reference in new issue