Skip to content
This repository has been archived by the owner on Dec 29, 2022. It is now read-only.

Fail to train VGG loss #30

Open
yzxing87 opened this issue Oct 9, 2020 · 1 comment
Open

Fail to train VGG loss #30

yzxing87 opened this issue Oct 9, 2020 · 1 comment

Comments

@yzxing87
Copy link

yzxing87 commented Oct 9, 2020

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.

@Richerhooders
Copy link

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants