Skip to content

Which part of the computation has been reduced? #6

@Vincent-Hoo

Description

@Vincent-Hoo

As stated in the paper, the proposed convnet-aig can reduce computational cost. But after checking the source codes, I found each layer must be executed. The way you skip a single layer is by multiplying a mask (0 or 1) to the output. If that is the case, which part of the convnet-aig can reduce the computation?
The code below is cited from convnet_aig.py, line 152, where out is the output of a layer, w[:,1] is the mask.

out = self.shortcut(x) + out * w[:,1].unsqueeze(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