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 6
    • Issues 6
    • 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
  • Merge requests
  • !22

Open
Created Apr 06, 2022 by yoon_jaeseok@yoon_jaeseok
  • Report abuse
Report abuse

Fix nle version at requirement.txt

  • Overview 0
  • Commits 1
  • Changes 1

Because current nle module version is up to 0.8.x,

I got some issue with shape of model like below when I test with test_submission.py.

So, I think we need to set version to 0.7.3 like docker environment you provide.

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 "/home/jake/workspace/neurips-2021-the-nethack-challenge/agents/torchbeast_agent.py", line 20, in __init__
    self.model = load_model(MODEL_DIR, self.device)
  File "/home/jake/workspace/neurips-2021-the-nethack-challenge/nethack_baselines/torchbeast/models/__init__.py", line 55, in load_model
    model.load_state_dict(checkpoint_states["model_state_dict"])
  File "/home/jake/anaconda3/envs/nle_challenge/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1406, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for BaselineNet:
	size mismatch for policy_logits_mask: copying a param with shape torch.Size([113]) from checkpoint, the shape in current model is torch.Size([121]).
	size mismatch for policy.weight: copying a param with shape torch.Size([113, 256]) from checkpoint, the shape in current model is torch.Size([121, 256]).
	size mismatch for policy.bias: copying a param with shape torch.Size([113]) from checkpoint, the shape in current model is torch.Size([121]).
Edited Apr 06, 2022 by yoon_jaeseok
Assignee
Assign to
Reviewer
Request review from
Time tracking
Source branch: master