stereo-magnification icon indicating copy to clipboard operation
stereo-magnification copied to clipboard

Fail to train VGG loss

Open yzxing87 opened this issue 4 years ago • 1 comments

Hi,

I can sucessfully train the pixel loss but failed at VGG loss (output images are black). The command I used to train VGG loss is

python train.py --cameras_glob=RealEstate10K/sample_dataset/camera_params/train/*txt \ --image_dir=RealEstate10K/sample_dataset/train --experiment_name=debug_vgg_loss --which_loss=vgg --batch_size=4 \ --vgg_model_file=imagenet-vgg-verydeep-19.mat --learning_rate=5e-5

The loss seems to descent normally. During the training process, most output images visualized by tensorboard are black while some are normal. I have tried to adjust different learning rates (1e-4, 5e-5, etc.) but it is still not working. Is there any special guideline to train the vgg loss? Thanks.

yzxing87 avatar Oct 09 '20 03:10 yzxing87

I have the same problem as you. Have you solved it?

Richerhooders avatar Dec 22 '22 02:12 Richerhooders