Skip to content

Bug? #11

Open
Open
Bug?#11
@iamhankai

Description

@iamhankai
iamhankai
opened on Oct 23, 2020

convnet-aig/gumbelmodule.py

Line 44 in 47633cd

soft_samples = F.softmax(gumble_trick_log_prob_samples / temperature, dim)

The dim here should be changed to -1?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions