pytorch1.1.0-python3.6-CUDA9.0-以MNIST为例子,使用最新的efficientnet,分类数目改成10
注意,该网络改变了两个东西以适应mnistdefload_pretrained_weights(model,model_name):"""Loadspretrainedweights,anddownloadsifloadingforthefirsttime."""state_dict=model_zoo.load_url(url_map[model_name])pretrained_dict={k