955a4eaa4c31782022f25f80781fffe62b3fa874,examples/library/minimal_transformer_training.py,,main,#,211

Before Change

  model.initialize(data_config)

  checkpoint_manager = tf.train.CheckpointManager(checkpoint, args.model_dir, max_to_keep=5)
  if checkpoint_manager.latest_checkpoint is not None:
    tf.get_logger().info("Restoring parameters from %s", checkpoint_manager.latest_checkpoint)
    checkpoint.restore(checkpoint_manager.latest_checkpoint)

  if args.run == "train":
    train(args.src, args.tgt, checkpoint_manager)
  elif args.run == "translate":
    translate(args.src)

After Change

      }
  }

  model = onmt.models.TransformerBase()
  runner = onmt.Runner(model, config, auto_config=True)

  if args.run == "train":
    runner.train()
  elif args.run == "translate":

Instances