Pytorch丨Pytorch相关资源整理

安装

pip install torch===1.4.0 torchvision===0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple -f https://download.pytorch.org/whl/torch_stable.html
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

日常操作

• pytorch入坑一 | Tensor及其基本操作
• 【pytorch】图像基本操作
• 权重初始化正则化
• 图像分类：数据增强（Pytorch版）
• [日常] PyTorch 预训练模型，保存，读取和更新模型参数以及多 GPU 训练模型
• pytorch中查看gpu信息

Bug

• Pytorch在训练过程中常见的问题

Tensorboard可视化

• PyTorch 自带 TensorBoard 使用教程

• Using TensorBoard with PyTorch 1.1 or 1.2

• Pytorch训练可视化(TensorboardX)

Hook

• pytorch使用hook打印中间特征图、计算网络算力等
• 半小时学会 PyTorch Hook