From 1aafbf91b312daca148d0b976a795191c5e22ae9 Mon Sep 17 00:00:00 2001 From: JianHe0628 <jl02958@surrey.ac.uk> Date: Tue, 3 Dec 2024 17:07:40 +0000 Subject: [PATCH] debug --- hamer/models/hamer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/hamer/models/hamer.py b/hamer/models/hamer.py index 9b16c50..2add59a 100644 --- a/hamer/models/hamer.py +++ b/hamer/models/hamer.py @@ -50,8 +50,10 @@ class HAMER(pl.LightningModule): # Instantiate MANO model mano_cfg = {k.lower(): v for k,v in dict(cfg.MANO).items()} - if self.mesh: - self.mano = MANO(**mano_cfg) + print('MANO IS HERE') + self.mano = MANO(**mano_cfg) + + print('MANO DONE') # Buffer that shows whetheer we need to initialize ActNorm layers self.register_buffer('initialized', torch.tensor(False)) -- GitLab