pytorch-vgg-cifar10
pytorch-vgg-cifar10 copied to clipboard
This is the PyTorch implementation of VGG network trained on CIFAR10 dataset
I need Vgg16 pretrained model. but there is error like this. RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.features.0.bias", "module.features.0.weight", "module.features.2.bias", "module.features.2.weight", "module.features.5.bias", "module.features.5.weight", "module.features.7.bias", "module.features.7.weight",...
Please pardon my poor English. And I wonder what's the version of pytorch and torchvision. I have met some error as follow: `RuntimeError: not allowed to set torch.backends.cudnn flags after...
Hello, I want to know if the data is divided equally among all workers?
I just run with ./run.sh and got nan loss after a few steps. Here is the printed log: (base) root@For-Judy-And-Ian:~/pytorchProjects/pytorch-vgg-cifar10-master# ./run.sh python main.py --arch=vgg11 --save-dir=save_vgg11 |& tee -a log_vgg11 Files...
I feel confused about the code in main function as follows : `model.features = torch.nn.DataParallel(model.features) ` May I ask that why not giving the whole model to DataParallel ?
I use the main.py to train cifar10 ,the result is incredible, the test acc is100% or 99.9%,