Skip to content

Commit f26e4ba

Browse files
authored
Update get_start_model.rst
1 parent 6b7f1f8 commit f26e4ba

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

docs/user/get_start_model.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,9 @@ Print model information
158158
# (dropout_2): Dropout(keep=0.8, name='dropout_2')
159159
# (dense_2): Dense(n_units=10, relu, in_channels='800', name='dense_2')
160160
# )
161+
162+
import pprint
163+
pprint.pprint(MLP.config) # print the model architecture
161164
162165
Get specific weights
163166
=======================

0 commit comments

Comments
 (0)