diff --git a/hamer/models/hamer.py b/hamer/models/hamer.py index 9b16c50d9f8472e08530871b8b2cfac33729d1ae..2add59af8b8c1d3b629165def88a690c0144ca02 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))