kd.from_xid.get_element_spec

kd.from_xid.get_element_spec#

kauldron.utils.from_xid.get_element_spec(
xid: int,
wid: int = 1,
) → PyTree[enp.ArraySpec][source]

Returns the element_spec of the train dataset of an xmanager experiment.

This allow to initialize an existing model without having to load an actual dataset, by using kd.data.tf.ElementSpecDataset.

Parameters:

  • xid – The XID of the experiment to load the config from.

  • wid – ID of the worker to load the config from. Defaults to 1.

Returns:

The element_spec of the train dataset.