NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Other
8.58k stars 1.2k forks source link

Floating point exception (core dumped) when training with 1 channel image #37

Open fniroui opened 5 years ago

fniroui commented 5 years ago

I believe this is caused by attempting to normalize the 1 channel input images by mean and std which have 3 channels. base_dataset.py line 123.