.gitignore
.pre-commit-config.yaml
CITATION.cff
LICENSE
README.md
requirements-dev.txt
requirements.txt
setup.cfg
setup.py
.github/codecov.yml
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/test.yml
tests/__init__.py
tests/conftest.py
tests/exceptions_test.py
tests/gpu_test.py
tests/half_precision_test.py
tests/jupyter_test.ipynb
tests/torchinfo_test.py
tests/torchinfo_xl_test.py
tests/fixtures/__init__.py
tests/fixtures/genotype.py
tests/fixtures/models.py
tests/fixtures/tmva_net.py
tests/test_output/ascii_only.out
tests/test_output/autoencoder.out
tests/test_output/basic_summary.out
tests/test_output/batch_size_optimization.out
tests/test_output/container.out
tests/test_output/containers.out
tests/test_output/custom_parameter.out
tests/test_output/device.out
tests/test_output/dict_input.out
tests/test_output/dict_parameters_1.out
tests/test_output/dict_parameters_2.out
tests/test_output/dict_parameters_3.out
tests/test_output/empty_module.out
tests/test_output/eval_order_doesnt_matter.out
tests/test_output/exception.out
tests/test_output/fasterrcnn.out
tests/test_output/frozen_layers.out
tests/test_output/genotype.out
tests/test_output/google.out
tests/test_output/input_size_possibilities.out
tests/test_output/input_tensor.out
tests/test_output/jit.out
tests/test_output/linear_model_half.out
tests/test_output/lstm.out
tests/test_output/lstm_custom_batch_size.out
tests/test_output/lstm_half.out
tests/test_output/mixed_trainable_parameters.out
tests/test_output/model_with_args.out
tests/test_output/module_dict.out
tests/test_output/multiple_input_tensor_args.out
tests/test_output/multiple_input_tensor_dict.out
tests/test_output/multiple_input_tensor_list.out
tests/test_output/multiple_input_types.out
tests/test_output/namedtuple.out
tests/test_output/pack_padded.out
tests/test_output/parameter_list.out
tests/test_output/partial_jit.out
tests/test_output/pruned_adversary.out
tests/test_output/pruning.out
tests/test_output/recursive.out
tests/test_output/resnet152.out
tests/test_output/resnet18_depth_consistency.out
tests/test_output/resnet50.out
tests/test_output/return_dict.out
tests/test_output/reusing_activation_layers.out
tests/test_output/row_settings.out
tests/test_output/siamese_net.out
tests/test_output/single_input.out
tests/test_output/single_input_all_cols.out
tests/test_output/single_input_batch_dim.out
tests/test_output/single_input_half.out
tests/test_output/single_layer_network_on_gpu.out
tests/test_output/single_linear_layer.out
tests/test_output/string_result.out
tests/test_output/tmva_net_column_totals.out
tests/test_output/too_many_linear.out
tests/test_output/too_many_linear_plus_existing_hooks.out
tests/test_output/too_many_relus.out
tests/test_output/uninitialized_tensor.out
torchinfo/__init__.py
torchinfo/enums.py
torchinfo/formatting.py
torchinfo/layer_info.py
torchinfo/model_statistics.py
torchinfo/py.typed
torchinfo/torchinfo.py
torchinfo.egg-info/PKG-INFO
torchinfo.egg-info/SOURCES.txt
torchinfo.egg-info/dependency_links.txt
torchinfo.egg-info/top_level.txt