當前位置:成語大全網 - 書法字典 - Torch.load()用法

Torch.load()用法

torch.load()的作用是從文件中加載用torch.save()保存的對象。

api:

參數:

默認加載方式:使用cpu加載cpu訓練的模型或使用gpu調用gpu訓練的模型;

在cpu上加載所有張量:

使用函數將所有張量加載到CPU中(適合GPU訓練的模型加載到CPU中):

將所有張量加載到第壹個GPU中(在CPU中訓練並在GPU中加載):

從GPU 1到GPU 0的映射張量(第壹次GPU訓練,第二次GPU加載):

根據您的設備,將張量加載到您的當前設備中: