dotnet / dotnet/TorchSharpExamples

[Suggestion] Improving TorchSharp experience in Notebook with Torchsharp.Summary

Aperta
#29 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Jupyter Notebook
Stelle
175
Fork
61
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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
==========================================================================================
```

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.