def _forward_once_for_flops(self, model):
    self.logger.info('Training parallel: Forward one batch for the flops information')
    (inputs, _) = next(iter(self.train_queue))
    model(inputs.to(self.device))
def _forward_once_for_flops(self, model):
    self.logger.info('Training parallel: Forward one batch for the flops information')
    (inputs, _) = next(iter(self.train_queue))
    model(inputs.to(self.device))