| 46 | config.local_encoder_heads = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.n_heads"); |
| 47 | config.local_encoder_window_size = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.window_size"); |
| 48 | config.local_encoder_max_seq_len = yaml::require_int(parsed, "model.init_args.local_encoder.init_args.max_seq_len"); |
| 49 | config.quantizer_input_dim = yaml::require_int(parsed, "model.init_args.local_quantizer.init_args.input_dim"); |
| 50 | config.quantizer_output_dim = yaml::require_int(parsed, "model.init_args.local_quantizer.init_args.output_dim"); |
| 51 | config.quantizer_levels = yaml::require_list_int(parsed, "model.init_args.local_quantizer.init_args.levels"); |
| 52 | config.codebook_size = compute_codebook_size(config.quantizer_levels); |
| 53 | |
| 54 | config.global_encoder_input_channels = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.input_channels"); |
| 55 | config.global_encoder_output_channels = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.output_channels"); |
| 56 | config.global_encoder_layers = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.num_layers"); |
| 57 | config.global_encoder_dim = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.dim"); |
| 58 | config.global_encoder_intermediate_dim = yaml::require_int(parsed, "model.init_args.global_encoder.init_args.intermediate_dim"); |
| 59 | |
| 60 | config.wave_prenet_dim = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.dim"); |
| 61 | config.wave_prenet_output_dim = yaml::require_int(parsed, "model.init_args.wave_prenet.init_args.output_dim"); |
no test coverage detected