Skip to content

Commit

Permalink
Fix wrong variable name
Browse files Browse the repository at this point in the history
  • Loading branch information
moehanabi committed Nov 10, 2024
1 parent db55f72 commit 536b237
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def def_torch_weight_reshape(weight):
# Step1: save pytorch model parameters to a.pth
# On the first run, uncomment lines 90 and 91.
# b = B()
# torch.save(a.state_dict(), 'a.pth')
# torch.save(b.state_dict(), 'a.pth')

a = A()
# Step2: Converts pytorch a.pth to the model parameter format of tensorlayerx
Expand Down

0 comments on commit 536b237

Please sign in to comment.