Get weights and bias from the saved model #2402
Answered
by
laggui
wangjiawen2013
asked this question in
Q&A
-
Hi, |
Beta Was this translation helpful? Give feedback.
Answered by
laggui
Oct 22, 2024
Replies: 1 comment
-
The parameters for each module are accessible via their respective field. For example, you can get the let weight = linear.weight.val();
let bias = linear.bias.unwrap().val(); |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
laggui
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
The parameters for each module are accessible via their respective field. For example, you can get the
Linear
weight and bias tensor values with: