@torch.no_grad()
def visualize_model(self):
    LOGGER.text('Visualizing architecture...', level=LoggerObserver.DEBUG)
    batch = next(iter(self.valloader))
    images = batch['inputs'].to(self.model.device)
    LOGGER.log([{'tag': 'Sanitycheck/analysis/architecture', 'value': self.model.model.get_model(), 'type': LoggerObserver.TORCH_MODULE, 'kwargs': {'inputs': images}}])
@torch.no_grad()
def visualize_model(self):
    LOGGER.text('Visualizing architecture...', level=LoggerObserver.DEBUG)
    batch = next(iter(self.valloader))
    images = batch['inputs'].to(self.model.device)
    LOGGER.log([{'tag': 'Sanitycheck/analysis/architecture', 'value': self.model.model.get_model(), 'type': LoggerObserver.TORCH_MODULE, 'kwargs': {'inputs': images}}])