Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add names and shapes of inputs and outputs to rten CLI output
This allows quickly inspecting the inputs and outputs of an RTen model. Example output for a BERT-like model: ``` Model summary: 2 inputs, 2 outputs, 81.5 M params Inputs input_ids: [batch_size, sequence_length] attention_mask: [batch_size, sequence_length] Outputs start_logits: [batch_size, sequence_length] end_logits: [batch_size, sequence_length] ```
- Loading branch information