Skip to content

Questions about the code #3

Open
@baumgach

Description

@baumgach

Dear Orobix team,

I am the first author of the VAGAN paper. I would like to thank you one more time for porting our code to PyTorch this has been immensely useful to the community working with these types of approaches! In fact, since TensorFlow 1 has become more or less obsolete people from my own team have started working with your code. This also motivated me to have a closer look at it. I have one potential issue and a question that I would like to bring to your attention:

  1. If I read the code correctly, you are re-using the last batch from the discriminator update for the generator update. While I don't think this would be a real issue in practice, I think to be completely correct and avoid unwanted dependencies between the two optimisation steps, a new batch should be used for the generator update. (https://github.com/orobix/Visual-Feature-Attribution-Using-Wasserstein-GANs-Pytorch/blob/master/src/train.py#L204)

  2. I do not understand why you set the requires_grad=True for the input images. This may be down to my lack of experience with PyTorch, though. What is the motivation for this?

Happy, to make a PR if you agree with my points.

Best wishes,
Christian

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