MCPcopy Create free account
hub / github.com/DragonisCV/RAM / test

Method test

ram/models/ram_base_model.py:119–128  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

117 self.model_ema(decay=self.ema_decay)
118
119 def test(self):
120 if hasattr(self, 'net_g_ema'):
121 self.net_g_ema.eval()
122 with torch.no_grad():
123 self.output = self.net_g_ema(self.lq)
124 else:
125 self.net_g.eval()
126 with torch.no_grad():
127 self.output = self.net_g(self.lq)
128 self.net_g.train()
129
130 def get_current_visuals(self):
131 out_dict = OrderedDict()

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected