torch.__future__

torch.__future__.set_overwrite_module_params_on_conversion(value)

設定在轉換 nn.Module 時，是否將新的 Tensor 指派給參數，而不是就地變更現有的參數。

啟用後，以下方法會將新的參數指派給模組

1. module.{device}()（例如 nn.Module.cuda()），用於在設備之間移動模組

2. module.{dtype}()（例如 nn.Module.float()），用於將模組轉換為不同的 dtype

3. nn.Module.to()

4. nn.Module.to_empty()

參數

value (bool) – 是否指派新的 Tensor。

torch.__future__.get_overwrite_module_params_on_conversion()

回傳在轉換 torch.nn.Module 時，是否要將新的 tensors 指定給參數，而不是就地變更現有的參數。預設為 False。

請參閱 set_overwrite_module_params_on_conversion() 以取得更多資訊。

回傳類型

bool

torch.__future__.set_swap_module_params_on_conversion(value)

設定在轉換 nn.Module 時，是否要使用 swap_tensors()，而不是設定 .data 以就地變更現有的參數；以及在將 state dict 載入 nn.Module 時，是否要使用 swap_tensors()，而不是使用 param.copy_(state_dict[key])。

注意

此函數的優先順序高於 get_overwrite_module_params_on_conversion()

啟用後，以下方法將就地交換現有的參數

1. module.{device}()（例如 nn.Module.cuda()），用於在設備之間移動模組

2. module.{dtype}()（例如 nn.Module.float()），用於將模組轉換為不同的 dtype

3. nn.Module.to()

4. nn.Module.to_empty()

5. nn.Module.load_state_dict()

當設定此選項時，load_state_dict() 的語意如下

1. 對於每個參數/緩衝區，其對應的 state_dict['key'] 會透過 module_load() 進行轉換 (即 res = param.module_load(state_dict['key']))

2. 如有必要，res 會被包裝在 Parameter 中

3. 模組中的參數/緩衝區將透過 swap_tensors() 與 res 進行交換

參數

value (bool) – 是否要使用 swap_tensors()。

torch.__future__.get_swap_module_params_on_conversion()

回傳在轉換 nn.Module 時，是否要使用 swap_tensors()，而不是設定 .data 以就地變更現有的參數。預設為 False。

請參閱 set_swap_module_params_on_conversion() 以取得更多資訊。

回傳類型

bool