Skip to content
Discussion options

You must be logged in to vote

Ok, I think I now know; it is simple enough:

for layer in model.children():
     for name, parameter in layer.named_parameters():
          'do whatever with them here, for example:'
           print( name, parametr.shape )

Sorry for asking trivialities; I'm still learning ;-) I hope this is useful to someone.

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@rusty1s
Comment options

Answer selected by errhernandez
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants