code: def forward(self, imgs, masks_enc, masks_pred):
      z = self.forward_context(imgs, masks_enc, masks_pred)
      h = self.forward_target(imgs, masks_enc, masks_pred)
      return z, h
  
file_path: lightly/examples/pytorch/ijepa.py
imports: import copy
  import torch
  import torchvision
  from torch import nn
  from torch.nn import functional
  from tqdm import tqdm
  from lightly.data.collate import IJEPAMaskCollator
  from lightly.models import utils
  from lightly.models.modules.ijepa import IJEPABackbone, IJEPAPredictor
  from lightly.transforms.ijepa_transform import IJEPATransform
label_desc: Use image node_id 4 for calling the IJEPA obj's underlying member method code with example usage: obj.forward(imgs, masks_enc, masks_pred) and returns: z,  h