cuvis_ai.utils.torch

Functions

extract_state_dict(state_dict[, format])
get_output_shape(input_shape, model)
guess_input_dimensionalty(model)	Inspects the first layer of a PyTorch model to guess if the model expects 1D (flattened), 2D (spatial), or 3D input data.
guess_state_dict_format(state_dict)

Classes

InputDimension(value)

An enumeration.