dotnet / dotnet/TorchSharpExamples
[Suggestion] Improving TorchSharp experience in Notebook with Torchsharp.Summary
- 主要言語
- Jupyter Notebook
- スター
- 175
- フォーク
- 61
- PR マージ指標
- 30日以内にマージされた PR はありません
説明
Is there interest to port
[torch-summary ](https://github.com/tyleryep/torchinfo)to TorchSharp.Summary
```python
from torchsummary import summary
model = ConvNet()
summary(model, (1, 28, 28))
```
So in TorchSharp notebook, we get:
```
==========================================================================================
Layer (type:depth-idx) Output Shape Param #
==========================================================================================
├─Conv2d: 1-1 [-1, 10, 24, 24] 260
├─Conv2d: 1-2 [-1, 20, 8, 8] 5,020
├─Dropout2d: 1-3 [-1, 20, 8, 8] --
├─Linear: 1-4 [-1, 50] 16,050
├─Linear: 1-5 [-1, 10] 510
==========================================================================================
Total params: 21,840
Trainable params: 21,840
Non-trainable params: 0
==========================================================================================
Input size (MB): 0.00
Forward/backward pass size (MB): 0.05
Params size (MB): 0.08
Estimated Total Size (MB): 0.14
==========================================================================================
```
コントリビューションガイド
このリポジトリのコントリビューションガイドは索引されていません
評価
この issue はまだ評価されていません。