colossalai.zero.sharded_model.utils

colossalai.zero.sharded_model.utils.col_model_deepcopy(sharded_model, other_model)[source]

copy param of the ShardedModelV2 to other_model. Note the other_model has to be the same as self.