Skip to content

Commit

Permalink
Update get_start_model.rst
Browse files Browse the repository at this point in the history
  • Loading branch information
zsdonghao committed Jun 28, 2019
1 parent f56dbb3 commit ad5ce65
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions docs/user/get_start_model.rst
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,53 @@ Print model information
import pprint
pprint.pprint(MLP.config) # print the model architecture
# {'inputs': '_inputlayer_1_node_0',
# 'model_architecture': [{'args': {'dtype': tf.float32,
# 'layer_type': 'normal',
# 'name': '_inputlayer_1',
# 'shape': [None, 784]},
# 'class': '_InputLayer',
# 'prev_layer': None},
# {'args': {'keep': 0.8,
# 'layer_type': 'normal',
# 'name': 'dropout_1'},
# 'class': 'Dropout',
# 'prev_layer': ['_inputlayer_1_node_0']},
# {'args': {'act': 'relu',
# 'layer_type': 'normal',
# 'n_units': 800,
# 'name': 'dense_1'},
# 'class': 'Dense',
# 'prev_layer': ['dropout_1_node_0']},
# {'args': {'keep': 0.8,
# 'layer_type': 'normal',
# 'name': 'dropout_2'},
# 'class': 'Dropout',
# 'prev_layer': ['dense_1_node_0']},
# {'args': {'act': 'relu',
# 'layer_type': 'normal',
# 'n_units': 800,
# 'name': 'dense_2'},
# 'class': 'Dense',
# 'prev_layer': ['dropout_2_node_0']},
# {'args': {'keep': 0.8,
# 'layer_type': 'normal',
# 'name': 'dropout_3'},
# 'class': 'Dropout',
# 'prev_layer': ['dense_2_node_0']},
# {'args': {'act': 'relu',
# 'layer_type': 'normal',
# 'n_units': 10,
# 'name': 'dense_3'},
# 'class': 'Dense',
# 'prev_layer': ['dropout_3_node_0']}],
# 'name': 'mlp',
# 'outputs': 'dense_3_node_0',
# 'version_info': {'backend': 'tensorflow',
# 'backend_version': '2.0.0-alpha0',
# 'save_date': None,
# 'tensorlayer_version': '2.1.0',
# 'training_device': 'gpu'}}
Get specific weights
=======================
Expand Down

0 comments on commit ad5ce65

Please sign in to comment.