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