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

Fail to train VGG loss #30

@yzxing87

Description

@yzxing87

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions