Skip to content

README.md

Latest commit

 

History

History
16 lines (9 loc) · 148 Bytes

README.md

File metadata and controls

16 lines (9 loc) · 148 Bytes
Raw

简单的迁移学习

配置：

python 3.6

torch 1.5.0

torchvision 0.6.0

cuda 10.2

运行：

修改数据集的路径之后

python main.py