TORCH_FORCE_WEIGHTS_ONLY_LOAD
|
如果設定為 [1, y, yes, true],則 torch.load 將使用 weights_only=True。即使在呼叫點傳遞了 weights_only=False,也會發生這種情況。有關更多文件,請參見 torch.load()。 |
TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD
|
如果設定為 [1, y, yes, true],則 torch.load 將使用 weights_only=False(如果在呼叫點未傳遞 weights_only 變數)。有關更多文件,請參見 torch.load()。 |
TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT
|
在某些條件下,autograd 執行緒在關閉時可能會掛起,因此我們不會無限期地等待它們關閉,而是依賴於預設設定為 10 秒的超時。此環境變數可用於設定超時時間(以秒為單位)。 |
TORCH_DEVICE_BACKEND_AUTOLOAD
|
如果設定為 1,則在執行 import torch 時,樹外後端擴充套件將自動匯入。 |