class_docstr: Policy object to apply parallelism per model
class_docstr_tok: ["Policy","object","to","apply","parallelism","per","model"]
docstr: Policy for class (module) replacement.
  
  Returns:
      Dict: Dictionary for class (module) replacement.
  
  Notes:
      The format of the dictionary object is as follows.
  
      dict:
          orig_class_name_1: reset_module_class_1,
          orig_class_name_2: reset_module_class_2,
          orig_class_name_3: reset_module_class_3,
          ...
          orig_class_name_4: reset_module_class_n
docstr_tok: ["Policy","for","class","(","module",")","replacement",".","Returns",":","Dict",":","Dictionary","for","class","(","module",")","replacement",".","Notes",":","The","format","of","the","dictionary","object","is","as","follows",".","dict",":","orig_class_name_1",":","reset_module_class_1",",","orig_class_name_2",":","reset_module_class_2",",","orig_class_name_3",":","reset_module_class_3",",","...","orig_class_name_4",":","reset_module_class_n"]
code: def replace_modules() -> Dict:
      """
      Policy for class (module) replacement.
  
      Returns:
          Dict: Dictionary for class (module) replacement.
  
      Notes:
          The format of the dictionary object is as follows.
  
          dict:
              orig_class_name_1: reset_module_class_1,
              orig_class_name_2: reset_module_class_2,
              orig_class_name_3: reset_module_class_3,
              ...
              orig_class_name_4: reset_module_class_n
      """
      return {}
  
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 3 for calling the Policy obj's underlying member method code with example usage: obj.replace_modules() and returns: dict