参考:https://www.bilibili.com/video/BV14E411H7Uw
在头文件中导入pytorch官方的resnet网络
import torchvision.models.resnet
然后打开resnet.py
在下述文件中下载网络,在train.py中载入
如果不想使用迁移学习,想自己训练,可以将下面标记的这段删去,
然后在net = resnet34()中传入参数
参考:https://www.bilibili.com/video/BV14E411H7Uw
在头文件中导入pytorch官方的resnet网络
import torchvision.models.resnet
然后打开resnet.py
在下述文件中下载网络,在train.py中载入
如果不想使用迁移学习,想自己训练,可以将下面标记的这段删去,
然后在net = resnet34()中传入参数