基本含义:

torchinfo 是一个方便的工具,用于总结 PyTorch 模型的结构(类似于 model.summary() 在 Keras 中的功能)

 第一部分:安装

conda install -c conda-forge torchinfo

第二部分:查看版本

python -c "import torchinfo; print(torchinfo.__version__)"

第三部分:卸载

conda remove torchinfo

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐