Skip to content

Commit e8b6c85

Browse files
committed
remove humanize dep
1 parent 000a8c8 commit e8b6c85

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

scripts/train.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616

1717
import configargparse
1818
import git
19-
import humanize
2019
import lightning as pl
2120
import numpy as np
2221
import psutil
@@ -942,7 +941,7 @@ def train(args):
942941

943942
model_lightning.to(device)
944943
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")
944+
logging.info(f"Model has {num_params / 1e6:.1f}M parameters")
946945

947946
if args.distributed:
948947
strategy = "ddp"

setup.cfg

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ train =
5959
wandb
6060
tensorboard
6161
configargparse
62-
humanize
6362
matplotlib
6463
kornia>=0.7.0
6564
gitpython

0 commit comments

Comments
 (0)