Closed jkkronk closed 3 years ago
I suspect that run_pretrained_unet_inference.py has a typo at line 57.
(output std + mean).cpu() ---> output = (output std + mean).cpu()
Thanks for raising this. Should be fixed now.
I suspect that run_pretrained_unet_inference.py has a typo at line 57.
(output std + mean).cpu() ---> output = (output std + mean).cpu()