← all repositories

ZhugeKongan/torch-template-for-deep-learning

A PyTorch deep learning template offering implementations of backbone CNNs, augmentation strategies, losses, attention modules, and visualization tools.

1.3k stars Python ML Frameworks
torch-template-for-deep-learning
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of numerous classical backbone CNN architectures along with common deep learning utilities. It includes data augmentation techniques like Cutout, Mixup, CutMix, and ShakeDrop, various loss functions, attention mechanisms, and visualization tools. The template comes with dataset loaders for multiple domains and a training script to facilitate model development.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.