This repo contains the code for training starmap with dropout, used in OrcVIO.
The following code in lib/models/hg.py is changed to add dropout layers
"""no dropout"""
# lin = nn.Sequential(nn.Conv2d(self.nFeats, self.nFeats, bias = True, kernel_size = 1, stride = 1),
# nn.BatchNorm2d(self.nFeats), self.relu)
"""add dropout"""
lin = nn.Sequential(nn.Conv2d(self.nFeats, self.nFeats, bias = True, kernel_size = 1, stride = 1),
nn.BatchNorm2d(self.nFeats), self.relu, nn.Dropout(0.5))