Skip to content

Commit 043b926

Browse files
committed
flip condition
1 parent 35848e9 commit 043b926

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

train_network.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1944,7 +1944,7 @@ def remove_model(old_ckpt_name):
19441944

19451945
dtype_to_use = torch.float64 if args.loss_related_use_float64 else torch.float32
19461946

1947-
if not args.disable_norm_metrics and not (args.full_bf16 or args.full_fp16):
1947+
if not args.disable_norm_metrics and (args.full_bf16 or args.full_fp16):
19481948
logger.warning("Unable to log gradients and model norms if full_bf16 or full_fp16, as requires float for quintiles.")
19491949

19501950
if args.full_bf16:

0 commit comments

Comments
 (0)