| 9 | |
| 10 | |
| 11 | class Logger(): |
| 12 | def __init__(self, cuda=False): |
| 13 | self.logger = logging.getLogger(__name__) |
| 14 | self.cuda = cuda |
| 15 | |
| 16 | def info(self, message, *args, **kwargs): |
| 17 | if (self.cuda and dist.get_rank() == 0) or not self.cuda: |
| 18 | self.logger.info(message, *args, **kwargs) |
| 19 | |
| 20 | def error(self, message, *args, **kwargs): |
| 21 | self.logger.error(message, *args, **kwargs) |
no outgoing calls
no test coverage detected