(save_path, flag_multigpus=False, l_level='INFO')
| 297 | |
| 298 | # put log into the dir |
| 299 | def setup_default_logging_wt_dir(save_path, flag_multigpus=False, l_level='INFO'): |
| 300 | |
| 301 | if flag_multigpus: |
| 302 | rank = dist.get_rank() |
| 303 | if rank != 0: |
| 304 | return |
| 305 | |
| 306 | tmp_timestr = time_str(fmt='%Y_%m_%d_%H_%M_%S') |
| 307 | new_log_path = os.path.join(save_path, tmp_timestr) |
| 308 | os.makedirs(new_log_path, exist_ok=True) |
| 309 | logger.add( |
| 310 | os.path.join(new_log_path, f'{tmp_timestr}.log'), |
| 311 | # os.path.join(new_log_path, f'training.log'), |
| 312 | level=l_level, |
| 313 | # format='{time:YYYY-MM-DD HH:mm:s} {file}[{line}] {level}: {message}', |
| 314 | format='{level}|{time:YYYY-MM-DD HH:mm:ss}: {message}', |
| 315 | # retention='30 days', |
| 316 | # rotation='30mb', |
| 317 | enqueue=True, |
| 318 | encoding='utf-8', |
| 319 | ) |
| 320 | return tmp_timestr |
| 321 | |
| 322 | |
| 323 | # - - - - - - - - - - - - - - - - - - - - - - - - - - - |
no test coverage detected