Method _visualize_batch

src/encoding/model.py:283–288

(self, step)

    @torch.no_grad()
    def _visualize_batch(self, step):
        bs = 2
        feat_maps = self.encode()
        for i in range(3):
            fm = feat_maps[i].detach().cpu().numpy()[0, 0]
            self.tb.add_figure(f"feat_map_{i}", draw_scalar_field2D(fm), global_step=step)
    
    @torch.no_grad()
    def evaluate(self):