code: def create_scheduler(optimizer):
      factor = (
          FLAGS.unroll_length
          * FLAGS.virtual_batch_size
          / FLAGS.total_steps
      )
      scheduler = torch.optim.lr_scheduler.LambdaLR(
          optimizer, lambda epoch: max(1 - epoch * factor, 0)
      )
      return scheduler
  
imports: import copy
  import dataclasses
  import getpass
  import logging
  import os
  import pprint
  import signal
  import socket
  import time
  from typing import Optional
  import coolname
  import hydra
  import moolib
  import omegaconf
  import torch
  import torch.nn
  import torch.nn.functional
  import wandb
  from moolib.examples.common import nest
  from moolib.examples.common import record
  from moolib.examples.common import vtrace
  from moolib.examples import common
  from ..atari import environment
  from ..atari import models
label_desc: Use image node_id 5 for calling a global function with example usage: create_scheduler(optimizer) and returns: scheduler