CoatNet-PyTorch-Lightning

This repository contains an implementation of CoatNet using PyTorch Lightning, allowing for efficient and scalable training.

CoatNet Overview

CoatNet integrates the strengths of convolutional neural networks (CNNs) and Transformers for image classification tasks. For more details, refer to the original paper: CoatNet: Marrying Convolution and Attention for All Data Sizes.

Features

• CoatNet model implementation
• Training routine using PyTorch Lightning
• Support for Distributed Data Parallel (DDP) training

Quick Start

To get started with training CoatNet on your dataset:

1. Clone this repository.
2. Install the required dependencies.
3. Run the training script with your dataset.

Requirements

• Python 3.x
• PyTorch
• PyTorch Lightning

For a detailed list of requirements, see requirements.txt.