GitHub - wangq95/pytorch_image_classification at e267e88a281e90ebe69373c7725a5aae96003894

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
data		data
models		models
script		script
.gitignore		.gitignore
README.md		README.md
config.py		config.py
train.py		train.py
train_dali.py		train_dali.py
utils.py		utils.py

Repository files navigation

Requirements

Pytorch 1.0
Python 3.6+
DALI

# Train A network with ImageNet64 memory data by dali
./script/data_to_memory.sh imagenet64
python train_dali.py --dataset ImageNet64 --data_path /userhome/temp_data/cifar10 --model_method proxyless_NAS --model_name proxyless_gpu