dotnet / dotnet/TorchSharpExamples

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

Abierto
#29 0 comentarios 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Jupyter Notebook
Estrellas
175
Forks
61
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

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

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.