main.py 文件源码

def main(_):
  config = create_config()
  trainer = Trainer(config)

  # Register signal handler
  def stop_training(signum, frame):
    trainer.stop_training()

  signal.signal(signal.SIGINT, stop_training)

  if config.profile:
    import cProfile as profile
    profile.runctx('trainer.train()', globals(), locals(), 'main.prof')
  else:
    trainer.train()