torch.utils.model_zoo.load_url の使用法

事前トレーニング済みモデルをロードするときは、pytorch に付属する model_zoo メソッドを使用して、URL アドレスでモデルをロードします。具体的な API は次のとおりです。

torch.utils.model_zoo.load_url(url, model_dir=None)

役割: 指定された URL にトーチシリアル化オブジェクトをロードします。
具体的には、指定された .pth ファイルの URL アドレスを介して、指定された .pth ファイルをダウンロードします。
パラメータ:
url (文字列) - オブジェクトをダウンロードする URL
model_dir (文字列、オプション) - オブジェクトを保存するディレクトリ オブジェクトが
model_dir にすでに存在する場合、逆シリアル化されて返されます。[つまり後述 - モデルがダウンロードされている場合]
詳細説明:
オブジェクトがすでにmodel_dirに存在する場合は、デシリアライズされて返されます。URL のファイル名部分は、ファイル名-.ext という命名規則に従う必要があります。ここで、 はファイルの内容の SHA256 ハッシュの最初の 8 桁以上です。ハッシュは、一意の名前を確保し、ファイルの内容を検証するために使用されます。model_dir のデフォルト値はTORCHHOME/models (TORCH_HOME/models) です。トルクHHOME / mod e l s 。TORCH_HOMEのデフォルトは ~/.torchですデフォルトのディレクトリは、$TORCH_MODEL_ZOO 環境変数を使用して上書きできます。

おすすめ

転載: blog.csdn.net/q4326rgeqaadfd/article/details/129132975