| 243 | } |
| 244 | |
| 245 | struct ggml_opt_params ggml_opt_default_params( |
| 246 | ggml_backend_sched_t backend_sched, |
| 247 | enum ggml_opt_loss_type loss_type) { |
| 248 | return { |
| 249 | /*backend_sched =*/ backend_sched, |
| 250 | /*ctx_compute =*/ nullptr, |
| 251 | /*inputs =*/ nullptr, |
| 252 | /*logits =*/ nullptr, |
| 253 | /*loss_type =*/ loss_type, |
| 254 | /*build_type =*/ GGML_OPT_BUILD_TYPE_OPT, |
| 255 | /*opt_period =*/ 1, |
| 256 | /*get_opt_pars =*/ ggml_opt_get_default_optimizer_params, |
| 257 | /*get_opt_pars_ud =*/ nullptr, |
| 258 | /*optimizer =*/ GGML_OPT_OPTIMIZER_TYPE_ADAMW, |
| 259 | }; |
| 260 | } |
| 261 | |
| 262 | static ggml_tensor * map_tensor(std::map<ggml_tensor *, ggml_tensor *> & tensor_map, ggml_context * ctx, ggml_tensor * tensor) { |
| 263 | if (!tensor) { |
no outgoing calls