Pytorch入门实战 | 第P8周：YOLOv5-C3模块实现

**本次我将利用YOLOv5算法中的C3模块搭建网络，主要让大家先了解C3的结构，方便后续YOLOv5算法的学习。

🏡 我的环境：

• 语言环境：Python3.8
• 编译器：Jupyter Lab
• 数据集：天气识别数据集
• 深度学习环境：Pytorch
  • torch==1.12.1+cu113
  • torchvision==0.13.1+cu113

一、 前期准备

1. 设置GPU

如果设备上支持GPU就使用GPU,否则使用CPU

import torch
import torch.nn as nn
import torchvision.transforms as transforms