We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 000a8c8 commit e8b6c85Copy full SHA for e8b6c85
2 files changed
scripts/train.py
@@ -16,7 +16,6 @@
16
17
import configargparse
18
import git
19
-import humanize
20
import lightning as pl
21
import numpy as np
22
import psutil
@@ -942,7 +941,7 @@ def train(args):
942
941
943
model_lightning.to(device)
944
num_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
945
- logging.info(f"Model has {humanize.intword(num_params)} parameters")
+ logging.info(f"Model has {num_params / 1e6:.1f}M parameters")
946
947
if args.distributed:
948
strategy = "ddp"
setup.cfg
@@ -59,7 +59,6 @@ train =
59
wandb
60
tensorboard
61
configargparse
62
- humanize
63
matplotlib
64
kornia>=0.7.0
65
gitpython
0 commit comments