diff --git a/python/mxnet/gluon/block.py b/python/mxnet/gluon/block.py index f58b00dded2e..af54c66c42d8 100644 --- a/python/mxnet/gluon/block.py +++ b/python/mxnet/gluon/block.py @@ -656,10 +656,12 @@ def _summary_hook(block, _, outputs): trainable_params += summary[layer]['trainable'] shared_params += summary[layer]['shared'] print('='*80) - print('Total params: ' + str(total_params)) - print('Trainable params: ' + str(trainable_params)) - print('Non-trainable params: ' + str(total_params - trainable_params)) - print('Shared params: ' + str(shared_params)) + print('Parameters in forward computation graph, duplicate included') + print(' Total params: ' + str(total_params)) + print(' Trainable params: ' + str(trainable_params)) + print(' Non-trainable params: ' + str(total_params - trainable_params)) + print('Shared params in forward computation graph: ' + str(shared_params)) + print('Unique parameters in model: ' + str(total_params - shared_params)) print('-'*80) finally: for h in hooks: