-
Notifications
You must be signed in to change notification settings - Fork 124
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add the "groups" column. Add test_groups(). Add groups field to single_input_all_cols.out Add Groups field to parameter_list.out * Move the GROUPS field so that it comes after KERNEL_SIZE. This changes the order of the ColumnSettings enumeration. --------- Co-authored-by: Andrew Lavin <[email protected]>
- Loading branch information
1 parent
6e0668a
commit 29166cc
Showing
9 changed files
with
59 additions
and
19 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -167,6 +167,7 @@ Args: | |
"num_params", | ||
"params_percent", | ||
"kernel_size", | ||
"groups", | ||
"mult_adds", | ||
"trainable", | ||
) | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
================================================================================================================================================================ | ||
Layer (type:depth-idx) Kernel Shape Groups Input Shape Output Shape Param # Mult-Adds | ||
================================================================================================================================================================ | ||
Conv2d [3, 3] 4 [7, 16, 28, 28] [7, 32, 26, 26] 1,184 5,602,688 | ||
================================================================================================================================================================ | ||
Total params: 1,184 | ||
Trainable params: 1,184 | ||
Non-trainable params: 0 | ||
Total mult-adds (M): 5.60 | ||
================================================================================================================================================================ | ||
Input size (MB): 0.35 | ||
Forward/backward pass size (MB): 1.21 | ||
Params size (MB): 0.00 | ||
Estimated Total Size (MB): 1.57 | ||
================================================================================================================================================================ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,15 +1,15 @@ | ||
==================================================================================================================================================================================== | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Param # Param % Mult-Adds Trainable | ||
==================================================================================================================================================================================== | ||
ParameterListModel -- [100, 100] [100, 100] 30,000 100.00% -- True | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== | ||
Layer (type:depth-idx) Kernel Shape Groups Input Shape Output Shape Param # Param % Mult-Adds Trainable | ||
======================================================================================================================================================================================================== | ||
ParameterListModel -- -- [100, 100] [100, 100] 30,000 100.00% -- True | ||
======================================================================================================================================================================================================== | ||
Total params: 30,000 | ||
Trainable params: 30,000 | ||
Non-trainable params: 0 | ||
Total mult-adds (M): 0 | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== | ||
Input size (MB): 0.04 | ||
Forward/backward pass size (MB): 0.00 | ||
Params size (MB): 0.00 | ||
Estimated Total Size (MB): 0.04 | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,20 +1,20 @@ | ||
==================================================================================================================================================================================== | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Param # Param % Mult-Adds Trainable | ||
==================================================================================================================================================================================== | ||
SingleInputNet -- [7, 1, 28, 28] [7, 10] -- -- -- True | ||
├─Conv2d: 1-1 [5, 5] [7, 1, 28, 28] [7, 10, 24, 24] 260 1.19% 1,048,320 True | ||
├─Conv2d: 1-2 [5, 5] [7, 10, 12, 12] [7, 20, 8, 8] 5,020 22.99% 2,248,960 True | ||
├─Dropout2d: 1-3 -- [7, 20, 8, 8] [7, 20, 8, 8] -- -- -- -- | ||
├─Linear: 1-4 -- [7, 320] [7, 50] 16,050 73.49% 112,350 True | ||
├─Linear: 1-5 -- [7, 50] [7, 10] 510 2.34% 3,570 True | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== | ||
Layer (type:depth-idx) Kernel Shape Groups Input Shape Output Shape Param # Param % Mult-Adds Trainable | ||
======================================================================================================================================================================================================== | ||
SingleInputNet -- -- [7, 1, 28, 28] [7, 10] -- -- -- True | ||
├─Conv2d: 1-1 [5, 5] 1 [7, 1, 28, 28] [7, 10, 24, 24] 260 1.19% 1,048,320 True | ||
├─Conv2d: 1-2 [5, 5] 1 [7, 10, 12, 12] [7, 20, 8, 8] 5,020 22.99% 2,248,960 True | ||
├─Dropout2d: 1-3 -- -- [7, 20, 8, 8] [7, 20, 8, 8] -- -- -- -- | ||
├─Linear: 1-4 -- -- [7, 320] [7, 50] 16,050 73.49% 112,350 True | ||
├─Linear: 1-5 -- -- [7, 50] [7, 10] 510 2.34% 3,570 True | ||
======================================================================================================================================================================================================== | ||
Total params: 21,840 | ||
Trainable params: 21,840 | ||
Non-trainable params: 0 | ||
Total mult-adds (M): 3.41 | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== | ||
Input size (MB): 0.02 | ||
Forward/backward pass size (MB): 0.40 | ||
Params size (MB): 0.09 | ||
Estimated Total Size (MB): 0.51 | ||
==================================================================================================================================================================================== | ||
======================================================================================================================================================================================================== |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters