TrainingCode -- Classification

The code is used as quick deploy for training classification task.

Distribute Mode

The use of gpu can be setted in config.py DEVICE.

python train_val.py

The use of gpu can be setted in config.py DEVICE.

python -m torch.distributed.launch --nproc_per_node <number_of_gpu> train_val.py

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
utils		utils
.gitignore		.gitignore
README.md		README.md
ResNet.py		ResNet.py
arg_reader.py		arg_reader.py
config.py		config.py
data_transform.py		data_transform.py
data_transform_seg.py		data_transform_seg.py
load_data_mix_all_classify.py		load_data_mix_all_classify.py
load_data_train_val_classify.py		load_data_train_val_classify.py
load_model.py		load_model.py
test.py		test.py
train_val.py		train_val.py