First, you should move into the directory cd cls
You can run the following code to train on MNIST dataset
python main.py --data mnist --data_path <path/to/mnist> --model_type lenet5 --epochs 100 --lr-epochs 15 --lr 1e-3
You can run the following code to train on CIFAR-10 dataset
python main.py --data cifar10 --data_path <path/to/cifar10> --model_type vgg7 --epochs 150 --lr-epochs 30 --lr 0.1
You can run the following code to train on CIFAR-100 dataset
python main.py --data cifar100 --data_path <path/to/cifar100> --model_type vgg7 --epochs 150 --lr-epochs 30 --lr 0.1
Your can run the following code to train on ImageNet dataset
python main.py --data imagenet --data_path <path/to/imagenet> --model_type restnet18 --epochs 150 --lr-epochs 15 --lr 0.1
First, you should move into the directory cd det
Your can run the following code to train on VOC dataset
python train.py --data data/VOC.yaml --weights '' --cfg yolov5s_TWN.yaml