(opt)
| 15 | |
| 16 | |
| 17 | def init_tb_loggers(opt): |
| 18 | # initialize wandb logger before tensorboard logger to allow proper sync |
| 19 | if (opt['logger'].get('wandb') is not None) and (opt['logger']['wandb'].get('project') |
| 20 | is not None) and ('debug' not in opt['name']): |
| 21 | assert opt['logger'].get('use_tb_logger') is True, ('should turn on tensorboard when using wandb') |
| 22 | init_wandb_logger(opt) |
| 23 | tb_logger = None |
| 24 | if opt['logger'].get('use_tb_logger') and 'debug' not in opt['name']: |
| 25 | tb_logger = init_tb_logger(log_dir=osp.join(opt['root_path'], 'tb_logger', opt['name'])) |
| 26 | return tb_logger |
| 27 | |
| 28 | |
| 29 | def create_train_val_dataloader(opt, logger): |
no test coverage detected