An implementation of the ConvNeXt architecture built on the PyTorch-Lightning API
Project description
ConvNeXt-Lightning
An implementation of the ConvNeXt architecture built on the PyTorch-Lightning API. The base model code (forward passes and architecture) are from the FAIR repo here.
This library allows easy loading of an untrained ConvNeXt model from the PyTorch Lightning API. Additionally, it provides an ImageDataset(nn.Dataset) module for image classification tasks.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
convnextpl-0.0.1.tar.gz
(6.9 kB
view hashes)
Built Distribution
Close
Hashes for convnextpl-0.0.1-py2.py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 917e5bce16befa8b318e0cb9d39dccc22c8d9e8f75efd90e77c206b84d5ada49 |
|
| MD5 | c92a2ab38eea5d855fae9c0e036d46b1 |
|
| BLAKE2b-256 | be3ab97af3843f998eaddeb54c92414391549b5122a88778c6925f991293477a |
