ViT-TensorRT

View on PyPIReverse Dependencies (0)

0.1.0 vit_tensorrt-0.1.0-py3-none-any.whl

Wheel Details

Project: ViT-TensorRT
Version: 0.1.0
Filename: vit_tensorrt-0.1.0-py3-none-any.whl
Download: [link]
Size: 19017
MD5: b22a672226ad0a486372e52ccc3e113a
SHA256: 29928ba2ae2c4b27b7d0af1c3a2606c1fb6c8105892e936c8cebd234838c2dbe
Uploaded: 2024-10-12 14:47:29 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: ViT-TensorRT
Version: 0.1.0
Summary: A re-implementation of ViT containing utilities to convert to TensorRT engines and run in Triton.
Author-Email: KyleNewman97 <kyle[at]thenewmans.com.au>
License: MIT
Requires-Python: >=3.11
Requires-Dist: numpy (>=2.1.1)
Requires-Dist: pydantic (>=2.9.1)
Requires-Dist: torchvision (>=0.19.1)
Requires-Dist: scipy (>=1.14.1)
Requires-Dist: tqdm (>=4.66.5)
Requires-Dist: matplotlib (>=3.9.2)
Requires-Dist: tensorboard (>=2.17.1)
Requires-Dist: onnx (>=1.17.0)
Requires-Dist: torch (>=2.4.1)
Requires-Dist: nvidia-tensorrt (>=99.0.0)
Requires-Dist: pip (>=24.2)
Requires-Dist: tritonclient[http] (>=2.46.0)
Requires-Dist: tensorrt (==10.4.0)
Description-Content-Type: text/markdown
[Description omitted; length: 3483 characters]

WHEEL

Wheel-Version: 1.0
Generator: pdm-backend (2.4.2)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
vit_tensorrt-0.1.0.dist-info/METADATA sha256=WRr6ru4FVeI1Py8tQeBBOhJRbeJ5-8RJwC7HbtHItyE 4273
vit_tensorrt-0.1.0.dist-info/WHEEL sha256=pM0IBB6ZwH3nkEPhtcp50KvKNX-07jYtnb1g1m6Z4Co 90
vit_tensorrt-0.1.0.dist-info/entry_points.txt sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs 34
vit_tensorrt-0.1.0.dist-info/licenses/LICENSE sha256=t9U4hFoGZyY5OBw4r7xyOBSVrbzdYmaV4APgQSXvyDs 1067
vit_tensorrt/__init__.py sha256=6lYT8NCAiJ1yIBjWZuhzjlT-3w8qalFmiViJcq-nCm0 43
vit_tensorrt/analysis/__init__.py sha256=nv4ZufFktine8zQV1fRXKEYHQxGJ1UiiCnxr1-SjJ_U 78
vit_tensorrt/analysis/stats_calculator.py sha256=w-MqdBvByNsuEZNo37y6FnyxCMkiI3LT2wPCqNY1YNw 1488
vit_tensorrt/config/__init__.py sha256=uehjuTE9Nw9_tPIMUXCB48r9yN2ijncEBth4FeTe6yU 283
vit_tensorrt/config/attention_config.py sha256=m-od9i0g6f1_HiuDUkgOHaG2LBSz3FP1iPRQKgkZ-pQ 151
vit_tensorrt/config/encoder_config.py sha256=M9B4-luTSV2T5OV6WUawEv5NFPAXSwEttVzfYxqkagk 410
vit_tensorrt/config/mlp_config.py sha256=QE2P2vNmGKy9lMGB6dRCcVBRGoDRGAmmI0Xs-5avrvg 126
vit_tensorrt/config/train_config.py sha256=-NgrswJkHRpU7YHpJ-nIXdBkVCPBgBHBXrS247CnJw8 1230
vit_tensorrt/config/vit_config.py sha256=600FNEtDSEE4dpxlj4WYEvMfI40wNJStLaZAcvlVJME 385
vit_tensorrt/data/__init__.py sha256=YTs3SZaENgAbGAfaIMRXskF4TnE1iLFSF67Mj5kynwI 244
vit_tensorrt/data/vit_dataset.py sha256=vlrDMui-uTAkjYGTIOSbzmj9CHHUlJ8PjAaTmlebl98 4060
vit_tensorrt/data/vit_predict_image_transform.py sha256=LH0hE4NE37Am-A_Tlk1WX8S_5-eRSwdJWzv_nz3MNa0 564
vit_tensorrt/data/vit_train_image_transform.py sha256=TJNBdIF6AjdPeM3KHLe87Wtn3s8aPz3vPxH7Ky71Xpg 987
vit_tensorrt/model/__init__.py sha256=6z0SfSn7nIT2UBMiNO9N7fUz3uKvw-0uA0jrbzL0xUY 41
vit_tensorrt/model/encoder.py sha256=olEn7qsF8vQ1BFyCS2cvVKTEplInEIbdtHbiS0Ieux8 2560
vit_tensorrt/model/encoder_block.py sha256=vcFiYrW6uSnVVNEJGpYRqTw5NOR3RtJnimbsC7llYQE 3343
vit_tensorrt/model/mlp_block.py sha256=rB28iSIlfaQkJl9cJ579_9qk7kd084ZEDFOZJ8xUeKE 1037
vit_tensorrt/model/vit.py sha256=9PnjUp2hjhw_zI5NBLTQA7WkmWjDlS21SX6YthQ9ykY 16922
vit_tensorrt/utils/__init__.py sha256=n9gmJIAy9l6vWxWQaSrpzrVx3MHgLxnFYKZKMNbsGtA 63
vit_tensorrt/utils/meta_logger.py sha256=XKeK_lnAmjuEEn1VXDUUcYQZ9ksV3-X2n9N8dm4iyjk 1361
vit_tensorrt-0.1.0.dist-info/RECORD

entry_points.txt

[empty]