Luma Labs interview question

Debug pytorch training code for an mnist image denoiser