|
|
@@ -37,7 +37,7 @@ from utils.wandb_logging.wandb_utils import WandbLogger, check_wandb_resume |
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__) |
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# |
|
|
|
|
|
|
|
|
def train(hyp, opt, device, tb_writer=None): |
|
|
def train(hyp, opt, device, tb_writer=None): |
|
|
logger.info(colorstr('hyperparameters: ') + ', '.join(f'{k}={v}' for k, v in hyp.items())) |
|
|
logger.info(colorstr('hyperparameters: ') + ', '.join(f'{k}={v}' for k, v in hyp.items())) |
|
|
|