Exemplo n.º 1
0
 def train(self, mode=True):
     nn.Module.train(self, mode)
     VMRN.train(self, mode)
Exemplo n.º 2
0
 def _init_modules(self):
     SSD._init_modules(self)
     VMRN._init_modules(self)
Exemplo n.º 3
0
 def _init_weights(self):
     SSD._init_weights(self)
     VMRN._init_weights(self)
Exemplo n.º 4
0
 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)