Skip to content

Update the pretrained models #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Nov 2, 2022
Merged

Conversation

jimgao1
Copy link
Contributor

@jimgao1 jimgao1 commented Nov 2, 2022

The previous version of the pretrained models were corrupted and would raise the following error:

>>> import habitat
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/workspace/habitat/analyzer/habitat/__init__.py", line 4, in <module>
    from habitat.tracking.operation import OperationTracker
  File "/workspace/habitat/analyzer/habitat/tracking/operation.py", line 4, in <module>
    from habitat.analysis.trace import Trace
  File "/workspace/habitat/analyzer/habitat/analysis/trace.py", line 5, in <module>
    class Trace:
  File "/workspace/habitat/analyzer/habitat/analysis/trace.py", line 11, in Trace
    DefaultPredictor = Predictor()
  File "/workspace/habitat/analyzer/habitat/analysis/predictor.py", line 66, in __init__
    self.linear_pred = RuntimePredictor(
  File "/workspace/habitat/analyzer/habitat/analysis/mlp/mlp.py", line 131, in __init__
    self.load_state(model_path)
  File "/workspace/habitat/analyzer/habitat/analysis/mlp/mlp.py", line 134, in load_state
    checkpoint = torch.load(path)
  File "/opt/conda/lib/python3.8/site-packages/torch/serialization.py", line 795, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/opt/conda/lib/python3.8/site-packages/torch/serialization.py", line 1002, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.

These are the updated ones.

@jimgao1 jimgao1 assigned jimgao1 and unassigned jimgao1 Nov 2, 2022
@jimgao1 jimgao1 requested a review from anurlybayev November 2, 2022 20:17
Copy link
Contributor

@anurlybayev anurlybayev left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The .pth files are in Git-LFS correct?

@jimgao1
Copy link
Contributor Author

jimgao1 commented Nov 2, 2022

The .pth files are in Git-LFS correct?

Yeah.

@jimgao1 jimgao1 merged commit e92d95c into main Nov 2, 2022
@jimgao1 jimgao1 deleted the nov11-update-pretrained-models branch November 2, 2022 21:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants