Skip to content

Commit a51b233

Browse files
committed
fix typo
1 parent 36da15b commit a51b233

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

yolo/utils/logging_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -340,12 +340,12 @@ def custom_wandb_log(string="", level=int, newline=True, repeat=True, prefix=Tru
340340

341341
if 1:
342342
from yolo.utils.callbacks import TorchGlobals
343-
callbacks.append(TorchGlobals())
343+
callbacks.append(TorchGlobals(float32_matmul_precision='auto'))
344344
checkpoint_init_args = {
345345
'monitor': 'train_loss',
346346
'mode': 'min',
347347
'save_top_k': 5,
348-
'filename': '{epoch:04d}-{step:06d}-trainloss{train_loss.3f}.ckpt',
348+
'filename': '{epoch:04d}-{step:06d}-trainloss{train_loss:.3f}.ckpt',
349349
'save_last': True,
350350
}
351351
import lightning

0 commit comments

Comments
 (0)