Skip to content
This repository has been archived by the owner on Nov 22, 2022. It is now read-only.

Fix Incorrect State Dict Assumption #1326

Closed
wants to merge 1 commit into from

Conversation

ArmenAg
Copy link
Contributor

@ArmenAg ArmenAg commented Apr 19, 2020

Summary:
PyText currently assumes that the state_dict is a flat key,value ordered dict where the value is typed as tensor. This is not neccessarily true: https://pytorch.org/tutorials/beginner/saving_loading_models.html#saving-multiple-models-in-one-file

This diff fixes this assumption.

Differential Revision: D21100333

@facebook-github-bot facebook-github-bot added CLA Signed Do not delete this pull request or issue due to inactivity. fb-exported labels Apr 19, 2020
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D21100333

Summary:
Pull Request resolved: facebookresearch#1326

PyText currently assumes that the state_dict is a flat key,value ordered dict where the value is typed as tensor. This is not neccessarily true: https://pytorch.org/tutorials/beginner/saving_loading_models.html#saving-multiple-models-in-one-file

This diff fixes this assumption.

Differential Revision: D21100333

fbshipit-source-id: 9fb586c2462e48429f7814ab56130b9cc4ff03bb
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D21100333

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 942a49f.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed Do not delete this pull request or issue due to inactivity. fb-exported Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants