From 33b7757fcbebf2b47aef927b0664ca4eb161dfce Mon Sep 17 00:00:00 2001 From: MSXYZ-GenAI Date: Wed, 29 Apr 2026 14:03:11 +0300 Subject: [PATCH] Log model parameter counts --- VideoTAADLAA.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/VideoTAADLAA.py b/VideoTAADLAA.py index 046e07b..56bdbfc 100644 --- a/VideoTAADLAA.py +++ b/VideoTAADLAA.py @@ -153,7 +153,7 @@ class VideoTAADLAA: net.eval() n_params = sum(p.numel() for p in net.parameters()) - log.info(f"[DLAA] Model parameters: {n_params/1e6:.2f}M") + log.info(f"[DLAA] Main model parameters: {n_params / 1e6:.2f}M") self.net_cache[key] = net @@ -214,9 +214,12 @@ class VideoTAADLAA: net = net.float() net.eval() + n_params = sum(p.numel() for p in net.parameters()) + self.texture_net_cache[key] = net log.info("[DLAA] Loaded %s", os.path.basename(model_path)) + log.info(f"[DLAA] Texture model parameters: {n_params / 1e6:.2f}M") except Exception as e: if not self._texture_missing_warned: