[MuseCoCo] models.ARCH_MODEL_REGISTRY[state["args"].arch]

Open
#152 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
4/5
Estimated time
3-5 days
Newbie friendliness
25/100
Issue type
Bug
Clarity
Needs clarification
Activity status
Stale
Tech stack
python
Domain
machine-learning

Research direction

Start with interactive_dict_v5_1billion.py at cli_main, attributes, and the checkpoint-loading call, then trace fairseq/checkpoint_utils.py through load_checkpoint_to_cpu and _upgrade_state_dict. Reproduce the failure with the MuseCoCo checkpoint and determine why linear_transformer_lm_1billion is absent from the registry. Done means the checkpoint loads without this KeyError; the issue names no test or proposed fix.

Written by the indexing model from the issue text.

Description

i get a error like this

Traceback (most recent call last): File "interactive_dict_v5_1billion.py", line 430, in <module> cli_main() File "interactive_dict_v5_1billion.py", line 420, in cli_main attributes(args) File "interactive_dict_v5_1billion.py", line 199, in attributes models, _model_args = checkpoint_utils.load_model_ensemble( File "/home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fairseq/checkpoint_utils.py", line 253, in load_model_ensemble ensemble, args, _task = load_model_ensemble_and_task( File "/home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fairseq/checkpoint_utils.py", line 282, in load_model_ensemble_and_task state = load_checkpoint_to_cpu(filename, arg_overrides) File "/home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fairseq/checkpoint_utils.py", line 235, in load_checkpoint_to_cpu state = _upgrade_state_dict(state) File "/home/disk2/wangjunhao01/anaconda3/envs/MuseCoco/lib/python3.8/site-packages/fairseq/checkpoint_utils.py", line 439, in _upgrade_state_dict registry.set_defaults(state["args"], models.ARCH_MODEL_REGISTRY[state["args"].arch]) KeyError: 'linear_transformer_lm_1billion'

### Tasks
Dominant language
Python
Stars
5k
Forks
499
No merged PRs in 30d

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from microsoft/muzic

All issues in microsoft/muzic

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.