code: def main():
      # Reset global step because we incremented in previous loop
      tracker.set_global_step(0)
  
      for i in range(1, 401):
          tracker.add_global_step()
          loss = train()
          tracker.add(loss=loss)
          if i % 10 == 0:
              tracker.save()
          if i % 100 == 0:
              logger.log()
          time.sleep(0.02)
  
imports: import time
  import numpy
  from labml import tracker, logger
label_desc: Use image node_id 2 for calling a global function with example usage: main() without return types