diagrams/
data/
test/
wandb/
cifar10/
results/
models/__pycache__/
helpers/__pycache__/
.vscode/
