Low contrast outputs

[jordanjalles]

Training seems to work fine with the preview image in tensorboard getting sharper over time.

However, when I run with the -predict flag the evaluation, video frames, and realtime demo all show as a grey image with low contrast aberrations. During the saving processes I get a lot of warnings about " UserWarning {path}_ours.png is a low contrast image".

I'm not sure what to do from here.

[pureexe]

Could you show the full command line that you add -predict into it.

It seems like the weight doesn't load correctly. make sure that you have a weight save in runs/cd02

[jordanjalles]

I see the default checkpoint is 100 epochs and I had been stopping before that to see what results look like. That was probably my miss. I'll just need to train past the checkpoints or run with smaller checkpoint epochs. Thanks for the help!