Skip to content

GitLab

  • Menu
Projects Groups Snippets
    • Loading...
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in
  • neurips-2021-the-nethack-challenge neurips-2021-the-nethack-challenge
  • Project information
    • Project information
    • Activity
    • Labels
    • Planning hierarchy
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 4
    • Issues 4
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 3
    • Merge requests 3
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Monitor
    • Monitor
    • Metrics
    • Incidents
  • Packages & Registries
    • Packages & Registries
    • Package Registry
    • Infrastructure Registry
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • nethack
  • neurips-2021-the-nethack-challengeneurips-2021-the-nethack-challenge
  • Issues
  • #14

Closed
Open
Created Jul 06, 2021 by Ghost User@ghost

loading pre-trained checkpoint failed

@eric_hammy I set AGENT = TorchBeastAgent in submission_config.py, and MODEL_DIR = "./saved_models/torchbeast/pretrained_0.5B" in agents/torchbeast_agent.py. Then I ran $ python test_submission.py but it gives me the following error message. (commit id: 3f9ef7f1)

Traceback (most recent call last):
  File "test_submission.py", line 36, in <module>
    evaluate()
  File "test_submission.py", line 25, in evaluate
    agent = Agent(num_envs, batched_env.num_actions)
  File "/data/private/research/AgentLearning/nethack_challenge/agents/torchbeast_agent.py", line 26, in __init__
    self.model = load_model(MODEL_DIR, self.device)
  File "/data/private/research/AgentLearning/nethack_challenge/nethack_baselines/torchbeast/models/__init__.py", line 54, in load_model
    checkpoint_states = torch.load(flags.checkpoint, map_location=device)
  File "/root/anaconda3/envs/nle_challenge/lib/python3.8/site-packages/torch/serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/root/anaconda3/envs/nle_challenge/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load
    result = unpickler.load()
  File "/root/anaconda3/envs/nle_challenge/lib/python3.8/site-packages/omegaconf/basecontainer.py", line 108, in __setstate__
    key_type = d["_metadata"].key_type
KeyError: '_metadata'
Edited Jul 06, 2021 by Ghost User
Assignee
Assign to
Time tracking