Digital Immune

This repository contains training scripts for various models on different datasets. The models are trained using PyTorch and support distributed training using torch.distributed.

Training and Run Scripts

sEMG

The train_sEMG.py script is used for training models on sEMG datasets.

Usage

python train_sEMG.py

Image

The train_pic.py script is used for training models on image datasets.

Usage

python train_pic.py

MSA

The train_msa.py script is used for training models on MSA datasets.

Usage

python train_msa.py

MRI

The train_mri.py script is used for training models on MRI datasets.

Usage

python train_mri.py

Requirements

• Python 3.8+
• PyTorch 1.8+
• CUDA 10.2+

Installation

Clone the repository:

git clone https://github.com/IRMVLab/Digital_Immune.git
cd Digital_Immune

License

This project is licensed under the GPL-3.0 License.