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