mmengine.model.merge_dict¶

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``(``update is not traceable). It should use merge_dict to replace xxx.update.

Parameters: *args – dictionary needs to be merged.
Returns: Merged dict from args
Return type: dict