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

安装tqdm库、matplotlib库、fvcore库

tqdm/tqdm: A Fast, Extensible Progress Bar for Python and CLI
关于Vision Transformer的复现&其环境配置_第1张图片
关于Vision Transformer的复现&其环境配置_第2张图片
使用fvcore计算Pytorch中模型的参数数量以及FLOPs
关于Vision Transformer的复现&其环境配置_第3张图片

安装TensorBoard

直接 pip install tensorboard
关于Vision Transformer的复现&其环境配置_第4张图片
Pytorch 1.9.0 + Tensorboard 2.5.0可视化工具使用记录

在Pytorch下安装TensorBoard及使用

你可能感兴趣的:(环境配置,transformer,pytorch,深度学习)