class_docstr: Wrap `opt` in a lookahead optimizer
class_docstr_tok: ["Wrap","`","opt","`","in","a","lookahead","optimizer"]
code: def load_state_dict(self, sd):
      self.count = sd.pop("count")
      self.slow_weights = sd.pop("slow_weights")
      self.opt.load_state_dict(sd)
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 5 for calling the Lookahead obj's underlying member method code with example usage: obj.load_state_dict(sd) without return types