Shortcuts

杂项环境变量

变量

描述

TORCH_FORCE_WEIGHTS_ONLY_LOAD

如果设置为[1, y, yes, true],torch.load将使用weight_only=True。更多关于此的文档,请参见torch.load()

TORCH_AUTOGRAD_SHUTDOWN_WAIT_LIMIT

在某些情况下,autograd线程在关闭时可能会挂起,因此我们不会无限期地等待它们关闭,而是依赖于默认设置为10秒的超时。此环境变量可用于设置超时时间(以秒为单位)。