Skip to content

README.md

Latest commit

 

History

History
59 lines (39 loc) · 954 Bytes

README.md

File metadata and controls

59 lines (39 loc) · 954 Bytes

MID

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/MID.git
cd MID

License

This project is licensed under the GPL-3.0 License.