diff --git a/src/models/model_base.h b/src/models/model_base.h index 09f3b7340..32bd9a60a 100644 --- a/src/models/model_base.h +++ b/src/models/model_base.h @@ -14,6 +14,21 @@ enum struct usage { raw, training, scoring, translation, embedding }; YAML_REGISTER_TYPE(marian::models::usage, int) +// 'FASTOPT_REGISTER_TYPE' +#if FASTOPT +namespace marian { +namespace fastopt_helpers { + +template <> +struct As { + static marian::models::usage apply(const FastOpt& node) { + return static_cast(As::apply(node)); + } +}; +} // namespace fastopt_helpers +} // namespace marian +#endif + namespace marian { namespace models {