!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