dotnet / dotnet/TorchSharpExamples
[Suggestion] Improving TorchSharp experience in Notebook with Torchsharp.Summary
- Ngôn ngữ chính
- Jupyter Notebook
- Star
- 175
- Fork
- 61
- Chỉ số merge pull request
- Không có pull request nào được merge trong 30 ngày
Mô tả
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
==========================================================================================
```
Hướng dẫn đóng góp
Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này
Đánh giá
Issue này chưa được đánh giá.