| | |
|---|
| | | "and exclude_keys excludes keys of model states for the initialization." |
|---|
| | | "e.g.\n" |
|---|
| | | " # Load all parameters" |
|---|
| | | " --init_param some/where/model.pth\n" |
|---|
| | | " --init_param some/where/model.pb\n" |
|---|
| | | " # Load only decoder parameters" |
|---|
| | | " --init_param some/where/model.pth:decoder:decoder\n" |
|---|
| | | " --init_param some/where/model.pb:decoder:decoder\n" |
|---|
| | | " # Load only decoder parameters excluding decoder.embed" |
|---|
| | | " --init_param some/where/model.pth:decoder:decoder:decoder.embed\n" |
|---|
| | | " --init_param some/where/model.pth:decoder:decoder:decoder.embed\n", |
|---|
| | | " --init_param some/where/model.pb:decoder:decoder:decoder.embed\n" |
|---|
| | | " --init_param some/where/model.pb:decoder:decoder:decoder.embed\n", |
|---|
| | | ) |
|---|
| | | group.add_argument( |
|---|
| | | "--ignore_init_mismatch", |
|---|
| | |
|---|
| | | preprocess=iter_options.preprocess_fn, |
|---|
| | | max_cache_size=iter_options.max_cache_size, |
|---|
| | | max_cache_fd=iter_options.max_cache_fd, |
|---|
| | | dest_sample_rate=args.frontend_conf["fs"], |
|---|
| | | ) |
|---|
| | | cls.check_task_requirements( |
|---|
| | | dataset, args.allow_variable_data_keys, train=iter_options.train |
|---|
