Metadata-Version: 2.1
Name: torch-model-hub
Version: 0.1.9
Summary: Model hub for Pytorch.
Home-page: https://gitee.com/summry/torch-model-hub
Author: summy
Author-email: fkfkfk2024@2925.com
License: UNKNOWN
Keywords: Pytorch,AI,Machine learning,Deep learning,torch
Platform: UNKNOWN
Requires-Python: >=3.6
Description-Content-Type: text/markdown

Usage Sample
''''''''''''

.. code:: python

        import torch
        from model.model import TextCNN
        X = torch.randn(8, 10, 128)
        model = TextCNN(embed_dim=128, kernel_sizes=(2, 3, 4), cnn_channels=64, out_features=2)
        output = model(X)


