-
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 is_trainable column * test: testcase for is_trainable column * model which has fully, partial and non trainable modules * update tests that require all coloums to display is_trainable coloumn as well * docs: update README.md * fix type ignores and nits * Rename is_trainable to trainable * Calculate trainable in pre_hook * Fix readme Co-authored-by: Tyler Yep <tyler.yep@robinhood.com>
- Loading branch information
Showing
10 changed files
with
106 additions
and
27 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
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,18 +1,18 @@ | ||
=================================================================================================================== | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Param # Mult-Adds | ||
=================================================================================================================== | ||
ParameterListModel -- -- -- -- -- | ||
├─ParameterList: 1-1 -- -- -- 30,000 -- | ||
│ └─0 [100, 100] ├─10,000 | ||
│ └─1 [100, 200] └─20,000 | ||
=================================================================================================================== | ||
================================================================================================================================================================ | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Param # Mult-Adds Trainable | ||
================================================================================================================================================================ | ||
ParameterListModel -- -- -- -- -- True | ||
├─ParameterList: 1-1 -- -- -- 30,000 -- True | ||
│ └─0 [100, 100] ├─10,000 | ||
│ └─1 [100, 200] └─20,000 | ||
================================================================================================================================================================ | ||
Total params: 30,000 | ||
Trainable params: 30,000 | ||
Non-trainable params: 0 | ||
Total mult-adds (M): 0.00 | ||
=================================================================================================================== | ||
================================================================================================================================================================ | ||
Input size (MB): 0.04 | ||
Forward/backward pass size (MB): 0.00 | ||
Params size (MB): 0.12 | ||
Estimated Total Size (MB): 0.16 | ||
=================================================================================================================== | ||
================================================================================================================================================================ |
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 # Mult-Adds | ||
============================================================================================================================================ | ||
SingleInputNet -- -- -- -- -- | ||
├─Conv2d: 1-1 [5, 5] [7, 1, 28, 28] [7, 10, 24, 24] 260 1,048,320 | ||
├─Conv2d: 1-2 [5, 5] [7, 10, 12, 12] [7, 20, 8, 8] 5,020 2,248,960 | ||
├─Dropout2d: 1-3 -- [7, 20, 8, 8] [7, 20, 8, 8] -- -- | ||
├─Linear: 1-4 -- [7, 320] [7, 50] 16,050 112,350 | ||
├─Linear: 1-5 -- [7, 50] [7, 10] 510 3,570 | ||
============================================================================================================================================ | ||
================================================================================================================================================================ | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Param # Mult-Adds Trainable | ||
================================================================================================================================================================ | ||
SingleInputNet -- -- -- -- -- True | ||
├─Conv2d: 1-1 [5, 5] [7, 1, 28, 28] [7, 10, 24, 24] 260 1,048,320 True | ||
├─Conv2d: 1-2 [5, 5] [7, 10, 12, 12] [7, 20, 8, 8] 5,020 2,248,960 True | ||
├─Dropout2d: 1-3 -- [7, 20, 8, 8] [7, 20, 8, 8] -- -- -- | ||
├─Linear: 1-4 -- [7, 320] [7, 50] 16,050 112,350 True | ||
├─Linear: 1-5 -- [7, 50] [7, 10] 510 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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
============================================================================================================================================ | ||
Layer (type:depth-idx) Kernel Shape Input Shape Output Shape Trainable | ||
============================================================================================================================================ | ||
MixedTrainable -- -- -- Partial | ||
├─Conv1d: 1-1 [1] [1, 1, 1] [1, 1, 1] True | ||
├─Conv1d: 1-2 [1] [1, 1, 1] [1, 1, 1] Partial | ||
├─Conv1d: 1-3 [1] [1, 1, 1] [1, 1, 1] False | ||
├─Dropout: 1-4 -- [1, 1, 1] [1, 1, 1] -- | ||
============================================================================================================================================ | ||
Total params: 6 | ||
Trainable params: 3 | ||
Non-trainable params: 3 | ||
Total mult-adds (M): 0.00 | ||
============================================================================================================================================ | ||
Input size (MB): 0.00 | ||
Forward/backward pass size (MB): 0.00 | ||
Params size (MB): 0.00 | ||
Estimated Total Size (MB): 0.00 | ||
============================================================================================================================================ |
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