hookeai.model_architectures.procedures.model_summary

Get PyTorch model summary data.

Functions

get_n_model_parameters

Get number of parameters of PyTorch model.

get_model_summary

Get summary of PyTorch model.

Functions

get_model_summary(model[, input_data, ...])	Get summary of PyTorch model.
get_n_model_parameters(model)	Get number of parameters of PyTorch model.