Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

how to get pytroch model layer name

for name, layer in model.named_modules():
  print(name, layer)
Source by discuss.pytorch.org #
 
PREVIOUS NEXT
Tagged: #pytroch #model #layer
ADD COMMENT
Topic
Name
2+3 =