From 886eeb0547763e426bf629bf686d1fd1398dee52 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 10 Oct 2023 09:10:46 +0000 Subject: [PATCH] chore(format): run black on dev --- infer/modules/train/train.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/infer/modules/train/train.py b/infer/modules/train/train.py index b03d572..b81a0d2 100644 --- a/infer/modules/train/train.py +++ b/infer/modules/train/train.py @@ -116,7 +116,11 @@ def main(): children[i].join() -def run(rank, n_gpus, hps,): +def run( + rank, + n_gpus, + hps, +): global global_step if rank == 0: logger = utils.get_logger(hps.model_dir)