tinker.types.Datum

class tinker.types.Datum(StrictBase)

Fields:

• loss_fn_inputs – Dictionary mapping field names to tensor data
• model_input

convert_tensors(data)

Convert torch.Tensor and numpy arrays to TensorData in loss_fn_inputs during construction.

Returns: Any