r"""Sets megbrain log level Args: level: new log level Returns: original log level
(level)
| 181 | _imperative_rt_logger.set_log_handler(_megbrain_logger) |
| 182 | |
| 183 | def set_mgb_log_level(level): |
| 184 | r"""Sets megbrain log level |
| 185 | |
| 186 | Args: |
| 187 | level: new log level |
| 188 | |
| 189 | Returns: |
| 190 | original log level |
| 191 | """ |
| 192 | _megbrain_logger.setLevel(level) |
| 193 | if level == logging.getLevelName("ERROR"): |
| 194 | rst = _imperative_rt_logger.set_log_level( |
| 195 | _imperative_rt_logger.LogLevel.Error |
| 196 | ) |
| 197 | elif level == logging.getLevelName("INFO"): |
| 198 | rst = _imperative_rt_logger.set_log_level( |
| 199 | _imperative_rt_logger.LogLevel.Info |
| 200 | ) |
| 201 | else: |
| 202 | rst = _imperative_rt_logger.set_log_level( |
| 203 | _imperative_rt_logger.LogLevel.Debug |
| 204 | ) |
| 205 | return rst |
| 206 | |
| 207 | set_mgb_log_level(_default_level) |
| 208 |