Skip to content

Add support for torch.set_default_device when initializing model parameters - #1158

Open
fshp971 wants to merge 1 commit into
facebookresearch:mainfrom
fshp971:create-parameter-via-torch-empty
Open

Add support for torch.set_default_device when initializing model parameters#1158
fshp971 wants to merge 1 commit into
facebookresearch:mainfrom
fshp971:create-parameter-via-torch-empty

Replace `torch.Tensor` with `torch.empty` in `fairscale/nn/model_para…

37d0a0d
Select commit
Loading
Failed to load commit list.
Meta CLA / Meta CLA Check succeeded Aug 31, 2026 in 0s

Contributor License Agreement is valid!

View more details on Meta CLA