关于Vision Transformer的复现&其环境配置

安装tqdm库、matplotlib库、fvcore库

tqdm/tqdm: A Fast, Extensible Progress Bar for Python and CLI
请添加图片描述
请添加图片描述
使用fvcore计算Pytorch中模型的参数数量以及FLOPs
在这里插入图片描述

安装TensorBoard

直接 pip install tensorboard
在这里插入图片描述
Pytorch 1.9.0 + Tensorboard 2.5.0可视化工具使用记录

在Pytorch下安装TensorBoard及使用

猜你喜欢

转载自blog.csdn.net/Nismilesucc/article/details/120537296