pytorch加载默认模型位置
在使用pytorch之时,免不了要加载预习训练好的默认模型,如使用models.squeezenet1_1,此时会默认将模型加载在~/.torch/models/文件夹下,可以使用对环境变量的操作将其路径改变,以加载到指定文件夹。对于模型的默认路径,是通过环境变量TORCH_HOME进行操控。通过os控制命令行修改环境变量。os.environ['TORCH_HOME'] = '/media/sunkj/WorkAndStudy/pytorch/models/'再进行模型加载,即可。结果如下:r