lenet_mnist / src /trainer.py

Commit History

Fixed devices and demo
a0f925f

rzimmerdev commited on

feature: fixed training loop arguments for Lightning module
2262103

rzimmerdev commited on

Implemented training loops and logging functionalities with PyTorch Lightning
b358118

rzimmerdev commited on