Pytorch Project Template

Description:

A boilerplate to get started with PyTorch projects.

Accelerate: Easily train on multiple GPUs, TPUs, and mixed precision.
PyTorch Compile: Optimize your model with accelerate and TorchDynamoPlugin.
Logging: Track metrics with Weights & Biases.

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
src		src
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
.python-version		.python-version
Makefile		Makefile
README.md		README.md
inference.py		inference.py
train.py		train.py