معرفی شرکت ها

autonvtx-0.1

Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
Card image cap
تبلیغات ما

مشتریان به طور فزاینده ای آنلاین هستند. تبلیغات می تواند به آنها کمک کند تا کسب و کار شما را پیدا کنند.

مشاهده بیشتر
0.1
0.1

توضیحات

https://github.com/zasdfgbnm/autonvtx
ویژگی مقدار
سیستم عامل -
نام فایل autonvtx-0.1
نام autonvtx
نسخه کتابخانه 0.1
نگهدارنده []
ایمیل نگهدارنده []
نویسنده Xiang Gao
ایمیل نویسنده qasdfgtyuiop@gmail.com
آدرس صفحه اصلی https://github.com/zasdfgbnm/autonvtx
آدرس اینترنتی https://pypi.org/project/autonvtx/
مجوز MIT
# Install ``` pip install autonvtx ``` # Usage Write your model as usual and `autonvtx(model)` to your model: ```python import torch import autonvtx class Model(torch.nn.Module): def __init__(self): super().__init__() self.layer1 = torch.nn.Linear(5, 5) self.layer2 = torch.nn.Linear(5, 5) def forward(self, x): x = self.layer1(x) x = self.layer2(x) return x m = Model().cuda() autonvtx(m) input_ = torch.randn(1024, 5, device='cuda') torch.cuda.profiler.start() for _ in range(10): output = m(input_) torch.cuda.profiler.stop() ``` The screenshot for this would be: ![Screenshot 1](screenshot1.png) It also works with existing models: ```python import torch import torchvision import autonvtx m = torchvision.models.resnet50() autonvtx(m) input_ = torch.randn(10, 3, 224, 224) torch.cuda.profiler.start() for _ in range(10): output = m(input_) torch.cuda.profiler.stop() ``` The screenshot for this would be: ![Screenshot 2](screenshot2.png)


نیازمندی

مقدار نام
- torch


نحوه نصب


نصب پکیج whl autonvtx-0.1:

    pip install autonvtx-0.1.whl


نصب پکیج tar.gz autonvtx-0.1:

    pip install autonvtx-0.1.tar.gz