DeepSpeed
Disable torch.nn.init when counting parmeters in initializing PipelineModule
#5258
Open
Open
Disable torch.nn.init when counting parmeters in initializing PipelineModule #5258
tanconghui
wants to merge
2 commits into
deepspeedai:master
from
tanconghui:disable-init-in-PipelineModule
disable torch.nn.init when counting parameters in initializing Pipeli…
884b8500
Merge branch 'master' into disable-init-in-PipelineModule
d0003fc2
Login to write a write a comment.
Login via GitHub