Skip to content

Question regarding Equivariance loss #1

Description

@danushkv

Hey authors!!! Thank you for implementation of IEN in contract to steerable CNNs. My question is regarding the equivarince loss. I understand that one term which goes into the loss, is the GroupPooled output from the layer. But, what exactly is the other term?
Let's say x1 is an image from the dataset X, then the GroupPooled output from the layer i is the first term. How exactly is the other term here defined? lines found here

temp_rot = torch.cat((temp_rot, augmented_temp.unsqueeze(0)),0)
loss2.append(equivariance_loss(temparr[j][:tbs],temp_rot))

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