Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Add unique number of parameters to summary output in Gluon Block #12077

Merged
merged 2 commits into from
Aug 9, 2018
Merged

Add unique number of parameters to summary output in Gluon Block #12077

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
242eb92
add unique parameters in summary output
Aug 8, 2018
a4e1060
rebuild
Aug 8, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions python/mxnet/gluon/block.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down