class_docstr: Modified from Partial-FC
class_docstr_tok: ["Modified","from","Partial-FC"]
code: def forward(self, total_features, norm_weight):
      torch.cuda.current_stream().wait_stream(self.stream)
      logits = linear(total_features, norm_weight)
      return logits
  
file_path: insightface/recognition/vpl/vpl.py
imports: import logging
  import os
  import torch
  import torch.distributed
  from torch.nn import Module
  from torch.nn.functional import normalize, linear
  from torch.nn.parameter import Parameter
label_desc: Use image node_id 4 for calling the VPL obj's underlying member method code with example usage: obj.forward(total_features, norm_weight) and returns: logits