weiaicunzai / pytorch-cifar100

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)
4.22k stars 1.17k forks source link

cu101是表示多少版本。train.py运行完成后权重文件保存在哪里? #95

Open sex611 opened 1 week ago

sex611 commented 1 week ago

在环境pytorch1.6.0+cu101中的cu101是表示cuda版本是10.1吗? 运行完python test.py -net vgg16 -weights path_to_vgg16_weights_file 无法找到权重文件的路径,但在train.py执行完成后并没有出现权重文件,还是说train.py的权重文件保存在了哪里