Luma Labs Interview Question

Debug pytorch training code for an mnist image denoiser