tf_agents.networks.layer_utils.print_summary
Stay organized with collections
Save and categorize content based on your preferences.
Prints a summary of a model.
tf_agents.networks.layer_utils.print_summary(
model,
line_length=None,
positions=None,
print_fn=None,
expand_nested=False,
show_trainable=False,
layer_range=None
)
Args |
model
|
Keras model instance.
|
line_length
|
Total length of printed lines (e.g. set this to adapt the
display to different terminal window sizes).
|
positions
|
Relative or absolute positions of log elements in each line. If
not provided, defaults to [0.3, 0.6, 0.70, 1.] .
|
print_fn
|
Print function to use. It will be called on each line of the
summary. You can set it to a custom function in order to capture the
string summary. It defaults to print (prints to stdout).
|
expand_nested
|
Whether to expand the nested models. If not provided,
defaults to False .
|
show_trainable
|
Whether to show if a layer is trainable. If not provided,
defaults to False .
|
layer_range
|
List or tuple containing two strings, the starting layer name
and ending layer name (both inclusive), indicating the range of layers
to be printed in the summary. The strings could also be regexes instead
of an exact name. In this case, the starting layer will be the first
layer that matches layer_range[0] and the ending layer will be the
last element that matches layer_range[1] . By default (None ) all
layers in the model are included in the summary.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-04-26 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
{
"lastModified": "Last updated 2024-04-26 UTC.",
"confidential": False
}