def train(self, mode=True): nn.Module.train(self, mode) VMRN.train(self, mode)
def _init_modules(self): SSD._init_modules(self) VMRN._init_modules(self)
def _init_weights(self): SSD._init_weights(self) VMRN._init_weights(self)
def _init_modules(self): assert cfg.TRAIN.VMRN.RELCLS_GRAD, "No gradients are applied to relationship convolutional layers." SSD._init_modules(self) VMRN._init_modules(self)