code: def model():
      return nn.Conv2d(3, 10, kernel_size=1)
  
file_path: kornia/test/x/test_x.py
imports: import pytest
  from torch import nn
  from kornia.metrics import AverageMeter
  from kornia.x import EarlyStopping, ModelCheckpoint
  from kornia.x.utils import TrainerState
label_desc: Use image node_id 1 for calling a global function with example usage: model() and returns: nn