/SSL-Image-Classification

Semi supervised learning methods for solving the image classification problem on large scale

Primary LanguagePython

Run the porgram

python main.py --arch=conv_net --epochs=100
python main.py --arch=resnet18 --data_transforms=resnet_input_transform --epochs=100
python main.py --arch=cifar_convnet --dataset=cifar10 --data_transforms=cifar10_input_transform
python main.py --arch=conv_net --epochs=100