mmengine.model.merge_dict — mmengine 0.10.7 documentation (original) (raw)

mmengine.model.merge_dict(*args)[source]¶

Merge all dictionaries into one dictionary.

If pytorch version >= 1.8, merge_dict will be wrapped by torch.fx.wrap, which will make torch.fx.symbolic_trace skip trace merge_dict.

Note

If a function needs to be traced by torch.fx.symbolic_trace, but inevitably needs to use update method of dict``(``updateis not traceable). It should use merge_dict to replacexxx.update.

Parameters:

*args – dictionary needs to be merged.

Returns:

Merged dict from args

Return type:

dict