1 /**
2 * Copyright 2021-2023 Huawei Technologies Co., Ltd
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 #ifndef MINDSPORE_INCLUDE_API_CALLBACK_CKPT_SAVER_H
17 #define MINDSPORE_INCLUDE_API_CALLBACK_CKPT_SAVER_H
18
19 #include <cstddef>
20 #include <string>
21 #include <vector>
22 #include <memory>
23 #include "include/api/callback/callback.h"
24 #include "include/api/dual_abi_helper.h"
25
26 namespace mindspore {
27
28 class MS_API CkptSaver : public TrainCallBack {
29 public:
30 inline CkptSaver(int save_every_n, const std::string &filename_prefix);
31 virtual ~CkptSaver();
32
33 private:
34 CkptSaver(int save_every_n, const std::vector<char> &filename_prefix);
35 };
36
CkptSaver(int save_every_n,const std::string & filename_prefix)37 CkptSaver::CkptSaver(int save_every_n, const std::string &filename_prefix)
38 : CkptSaver(save_every_n, StringToChar(filename_prefix)) {}
39
40 } // namespace mindspore
41 #endif // MINDSPORE_INCLUDE_API_CALLBACK_CKPT_SAVER_H
42