Skip to content
Snippets Groups Projects
Commit 4d072390 authored by Rishi Sharma's avatar Rishi Sharma
Browse files

Fix state_dict -> state_dict()

parent 6ad8ca8d
No related branches found
No related tags found
No related merge requests found
......@@ -60,7 +60,7 @@ class PartialModel(Sharing):
shared_params = dict()
shared_params["order"] = self.model.state_dict().keys()
shapes = dict()
for k, v in self.model.state_dict.items():
for k, v in self.model.state_dict().items():
shapes[k] = v.shape.tolist()
shared_params["shapes"] = shapes
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment