Deep Learning (Pair Programming with ChapGPT)

Intro

Use ChatGPT to pair program deep learning script

Requirements

See requirements.txt

MNIST training loop

# Prompt:
# write a pytorch training pipeline for mnist, use tensorboard to visualize the images and loss

Code: mnist_train.py

Command

python3 train.py