Пример #1
0
 def train(self, mode=True):
     nn.Module.train(self, mode)
     VMRN.train(self, mode)
Пример #2
0
 def _init_modules(self):
     SSD._init_modules(self)
     VMRN._init_modules(self)
Пример #3
0
 def _init_weights(self):
     SSD._init_weights(self)
     VMRN._init_weights(self)
Пример #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)