keep network mode unchange when use summary api (#27754)

* keep summary mode unchange

* add no grad decorator
my_2.0rc
LielinJiang 4 years ago committed by GitHub
parent b9c7c66ea5
commit d84eb9b33f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -106,6 +106,12 @@ def summary(net, input_size, dtypes=None):
warnings.warn(
"Your model was created in static mode, this may not get correct summary information!"
)
in_train_mode = False
else:
in_train_mode = net.training
if in_train_mode:
net.eval()
def _is_shape(shape):
for item in shape:
@ -143,9 +149,13 @@ def summary(net, input_size, dtypes=None):
result, params_info = summary_string(net, _input_size, dtypes)
print(result)
if in_train_mode:
net.train()
return params_info
@paddle.no_grad()
def summary_string(model, input_size, dtypes=None):
def _all_is_numper(items):
for item in items:

Loading…
Cancel
Save