5 projects
torcheeg
TorchEEG is a library built on PyTorch for EEG signal analysis. TorchEEG aims to provide a plug-and-play EEG analysis tool, so that researchers can quickly reproduce EEG analysis work and start new EEG analysis research without paying attention to technical details unrelated to the research focus.
vision-transformer-pytorch
VisionTransformer implemented in PyTorch.
torch_tresnet
TResNet: High Performance GPU-Dedicated Architecture
torch_text_similarity
Implementations of models and metrics for semantic text similarity. Includes fine-tuning and prediction of models
torchcluster
Torchcluster is a python package for cluster analysis.