class_docstr: Policy object to apply parallelism per model
class_docstr_tok: ["Policy","object","to","apply","parallelism","per","model"]
docstr: Constructor of Policy class
  
  Args:
      layer (nn.Module): The layer to apply the policy to
docstr_tok: ["Constructor","of","Policy","class","Args",":","layer","(","nn.Module",")",":","The","layer","to","apply","the","policy","to"]
code: def __init__(self, layer: nn.Module) -> None:
      """
      Constructor of Policy class
  
      Args:
          layer (nn.Module): The layer to apply the policy to
      """
      super().__init__()
      self.layer = layer
  
file_path: parallelformers/parallelformers/policies/base/policy.py
imports: from abc import ABC, abstractmethod
  from dataclasses import dataclass
  from typing import Any, Dict, List, Type
  from torch import nn
label_desc: Use image node_id 1 to create a new Policy object from inherited base classes: ABC with example: obj = Policy(layer)