tfm.core.train_utils.cast_leaf_nested_dict

Cast the leaves of a dictionary with arbitrary depth in place.

d The dictionary to extract value from.
cast_fn The casting function.

A dictionray with the same structure as d.