Pytorch預訓練模型下載並加載(以VGG爲例)自定義路徑

簡述 通常來說,Pytorch用torchvision調用vgg之類的模型話,若是電腦在cache(Pytorch硬編碼的一個地址)(若是在環境變量中添加了TORCH_HOME 和TORCH_MODEL_ZOO的話,就是在這兩個位置的聯合的路徑下,好比TORCH_MODEL_ZOO\model)不然就是在TORCH_HOME\models或者是~/.torch/modelsgit 好比,個人就是C
相關文章
相關標籤/搜索