Pytorch實戰指南---使用Pytorch完成Kaggle上的經典比賽:Dogs vs Cats---updating
阿新 • • 發佈:2019-02-19
來源
1.檔案組織架構
├── checkpoints/
├── data/
│ ├── init.py
│ ├── dataset.py
│ └── get_data.sh
├── models/
│ ├── init.py
│ ├── AlexNet.py
│ ├── BasicModule.py
│ └── ResNet34.py
└── utils/
│ ├── init.py
│ └── visualize.py
├── config.py
├── main.py
├── requirements.txt
├── README.md
可以看到,幾乎每個資料夾下都有init
例如在data/資料夾下有init.py,則在main.py 中就可以
from data.dataset import DogCat
而如果在data/init.py中寫入
from .dataset import DogCat
則在main.py中就可以直接寫為:
from data import DogCat
或者
import data;
dataset = data.DogCat
相比於from data.dataset import DogCat更加便捷。